解决：could not load library libcudnn_cnn_train.so.8
could not load library libcudnn_cnn_train.so.8
·
训练模型时报下面错误：
could not load library libcudnn_cnn_train.so.8
出现错误的可能原因：
PyTorch 有自己的 CUDA 依赖项（包括 cuDNN），所以需要从库路径中移除本地安装的 cuDNN，让 PyTorch 加载它自己的库。
如果想使用本地安装的 CUDA 工具包，可以从源代码构建 PyTorch。
解决方法之一：
cd /usr/local/cuda-12.1/lib64
sudo rm -f libcudnn*
cd /usr/local/cuda-12.1/include
sudo rm -f cudnn*
参考文档：
https://discuss.pytorch.org/t/could-not-load-library-libcudnn-cnn-train-so-8-but-im-sure-that-i-have-set-the-right-ld-library-path/190277/2
一站式 AI 云服务平台
更多推荐
2
0- 0
已为社区贡献1条内容
所有评论(0)