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
完成!!!