yolov8训练出现Could not load library libcudnn_cnn_train.so.8 问题及解决方案

错误

使用python开始训练yolov8时突然弹出报错

Could not load library libcudnn_cnn_train.so.8. Error: /usr/local/cuda-12.1/lib64/libcudnn_cnn_train.so.8: undefined symbol: 后面一长串类似的报错提醒。

原因

具体原因是安装cuda的时候，按照教程将cudnn也安装到了系统cuda路径中，然后在python虚拟环境中使用pip install torch时会自动安装一个cudnn，两个发生了冲突。

解决办法

首先激活虚拟环境，使用pip list查看：

里面确实安装了cudnn，解决办法也很简单，既然产生了冲突，要么把系统里面的cudnn去掉（不推荐，如果部署其他的如Tensor rt时会出现问题，反而更麻烦）。要么把虚拟环境中的cudnn去掉（推荐），所以只需要执行如下命令：

pip uninstall nvidia-cudnn-cu12

根据自己的具体版本进行替换即可。