解决: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。

解决方法之一:

python 复制代码
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

相关推荐
StarPrayers.17 小时前
用 PyTorch 搭建 CIFAR10 线性分类器:从数据加载到模型推理全流程解析
人工智能·pytorch·python
Sherry Wangs18 小时前
显卡算力过高导致PyTorch不兼容的救赎指南
人工智能·pytorch·显卡
Sunhen_Qiletian21 小时前
卷积神经网络搭建实战(二)——基于PyTorch框架和本地自定义图像数据集的食物分类案例(附输入图片预测功能)
pytorch·分类·cnn
星期天要睡觉21 小时前
深度学习——基于 ResNet18 的图像分类训练
pytorch·python·机器学习
GitNohup1 天前
安装Anaconda和Pytorch
pytorch·anaconda
Blossom.1181 天前
把AI“绣”进丝绸:生成式刺绣神经网络让古装自带摄像头
人工智能·pytorch·python·深度学习·神经网络·机器学习·fpga开发
BB_CC_DD1 天前
在NVIDIA Jetson Orin NX (Ubuntu 22.04, JetPack 5.1, CUDA 11 cuDnn8) 上安装PyTorch 2
pytorch·深度学习·ubuntu
姜—姜1 天前
通过构建神经网络实现项目预测
人工智能·pytorch·深度学习·神经网络
MYX_3091 天前
第四章 神经网络的基本组件
pytorch·深度学习·神经网络·学习
西柚小萌新2 天前
【深入浅出PyTorch】--7.2.PyTorch可视化2
人工智能·pytorch·python