Jittor 提示没有找到 Cudnn

问答
,
1

运行jittor出现了以下错误，如何解决

file cudnn.h not found in ['/usr/local/cuda/include', '/usr/local/cuda/targets/x86_64-linux/include', '/usr/include']
1 个赞
2

1. 让 Jittor 为您安装 CUDA 环境

如果您没有安装 CUDA ，或者您的 CUDA 环境不完整（如没有安装 Cudnn），您可以运行以下命令使用 Jittor 内置工具为 Jittor 安装 CUDA。

python3 -m jittor_utils.install_cuda

通过此命令下载的 CUDA 会存储在 Jittor 的 Cache 目录下，不会影响系统的运行环境。

2. 如果您不想重新下载，并且安装了多个 CUDA 版本

您可以设置环境变量 nvcc_path 指定需要使用的 CUDA 版本下的 nvcc 路径，如

export nvcc_path=/usr/local/cuda-11.4/bin/nvcc
python3 -m jittor.test.test_cuda

或者

nvcc_path=/usr/local/cuda-11.4/bin/nvcc python3 -m jittor.test.test_cuda