Pytorch开发者提到:
The PyTorch binaries ship with their own CUDA dependencies (including cuDNN), so remove your locally installed cuDNN (temporarily) from the library path and let PyTorch load it’s own libs. If you want to use your locally installed CUDA toolkit you could build PyTorch from source.
翻译过来就是安装编译好的Pytorch会拥有自己的cuda依赖库(包括cudnn),所以需要删掉本地已经安装好的cudnn。
最终解决办法:
屏蔽掉本地cuda cudnn相关的文件(个人建议最好备份到指定目录下面,而不是直接删掉)
- 这个是删除操作:
cd /usr/local/cuda-12.1/lib64
sudo rm -f libcudnn*
cd /usr/local/cuda-12.1/include
sudo rm -f cudnn*
- 你可以将这些文件挪到指定目录。