cuda、cudnnn安装、卸载、查询

cuda

cuda卸载

sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl

cuda安装

cuda下载:CUDA Toolkit 11.5 Update 1 Downloads | NVIDIA Developer

sudo sh cuda_9.0.176_384.81_linux.run

注意:在安装10.0以上版本时,叉代表选定,取消差代表不进行安装;

修改环境变量

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
export PATH=$PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0
source ~/.bashrc

cuda版本查询

nvcc -V

cudnn

cudnn卸载

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

cudnn安装

cudnn下载地址:cuDNN Archive | NVIDIA Developer

$ tar -xzvf cudnn-9.0-linux-x64-v7.tgz
sudo cp cuda/include/cudnn* /usr/local/cuda/include/
 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
 
sudo chmod a+r /usr/local/cuda/include/cudnn.h
 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
#建立软链接
sudo chmod +r libcudnn.so.7.0.5
sudo ln -sf libcudnn.so.7.0.5 libcudnn.so.7  
sudo ln -sf libcudnn.so.7 libcudnn.so     
sudo ldconfig

cudnn版本查询

不知道从哪个版本开始,cudnn的版本号已经不在cudnn.h中了,而是单独有一个文件cudnn_version.h,因此在完整安装方法基础上,还需要在复制cudnn.h的同时也将cudnn_version.h一起复制到我们自己的cuda对应目录下(/usr/local/cuda/include)

#原指令
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

#新指令:
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
ldconfig

之后,遇到一个问题:

解决方法如下:

sudo ln -sf libcudnn_cnn_infer.so.8 libcudnn_cnn_infer.so
sudo ln -sf libcudnn_cnn_infer.so.8.0.4 libcudnn_cnn_infer.so.8

sudo ln -sf libcudnn_cnn_train.so.8 libcudnn_cnn_train.so
sudo ln -sf libcudnn_cnn_train.so.8.0.4 libcudnn_cnn_train.so.8

sudo ln -sf libcudnn_ops_infer.so.8.0.4 libcudnn_ops_infer.so.8
sudo ln -sf libcudnn_ops_infer.so.8 libcudnn_ops_infer.so

sudo ln -sf libcudnn_ops_train.so.8.0.4 libcudnn_ops_train.so.8
sudo ln -sf libcudnn_ops_train.so.8 libcudnn_ops_train.so

sudo ln -sf libcudnn_adv_infer.so.8.0.4 libcudnn_adv_infer.so.8
sudo ln -sf libcudnn_adv_infer.so.8 libcudnn_adv_infer.so

sudo ln -sf libcudnn_adv_train.so.8.0.4 libcudnn_adv_train.so.8
sudo ln -sf libcudnn_adv_train.so.8 libcudnn_adv_train.so

完成!!!

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值