一. 安装和卸载 cuda
1. 安装cuda
下载cuda:CUDA Toolkit Archive | NVIDIA Developer
sudo chmod 777 cuda_9.0.176_384.81_linux.run
sudo ./cuda_9.0.176_384.81_linux.run
事前安装过驱动,所以不安装。
在.bashrc的末尾添加环境变量:
exportPATH=/usr/local/cuda-9.0/bin:$PATH
exportLD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
使设置生效:source ~/.bashrc
验证cuda是否安装成功
cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
2. 卸载cuda
卸载原来的cuda9.0(不需要卸载显卡驱动):
sudo /usr/local/cuda-9.0/bin/cuda-uninstaller
卸载之后, /usr/local/cuda-9.0目录下任然有文件存在,这是cudnn文件,还需要将cuda-9.0文件删除干净:
sudo rm -rf /usr/local/cuda-9.0
3. 查看cuda版本
cat /usr/local/cuda/version.txt
二. 安装cudnn
1. 安装cudnn
下载对应版本的cuDNN:https://developer.nvidia.com/rdp/cudnn-download
tar -zxvf cudnn-9.2-linux-x64-v7.6.1.34.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
验证安装:
nvcc -V
或者查看cuDNN版本验证:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
2.卸载cudnn
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn