0
点赞
收藏
分享

微信扫一扫

安装和删除cudnn

诗尚凝寒 2024-09-24 阅读 31
人工智能

安装和删除cudnn

为什么?

删除cudnn就可以用yolov8

安装cudnn就可以用paddle

这两个不能互相兼容
至于原因嘛,好像就是安装的paddle是不带cudnn的,必须依赖系统的cudnn
但是安装的ultralytics是自带cudnn的,所以需要删掉系统的cudnn

删除cuDnn

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*

sudo rm -rf ./include/cudnn.h
sudo rm -rf ./lib64/libcudnn*

安装

sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

环境变量

添加.bashrc下的环境变量

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

cudnn路径:

cd /home/gsh/Downloads/cuda_cudnn_tensort/cudnn-linux-x86_64-8.9.7.29_cuda12-archive

举报

相关推荐

0 条评论