常用命令记录

CUDA安装后

vim ~/.bashrc

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

source ~/.bashrc

sudo ln -s /usr/local/cuda-xx.x /usr/local/cuda

更换cuda版本

sudo rm -rf /usr/local/cuda

sudo ln -s /usr/local/cuda-xx.x /usr/local/cuda

cudnn下载安装

下载链接：cuDNN Archive | NVIDIA Developer

解压及文件复制

tar -xvf cudnn-linux-...

cd cuda

sudo cp include/cu*.h /usr/local/cuda-xx.x/include

sudo cp lib/libcudnn* /usr/local/cuda-xx.x/lib64

授权

sudo chmod a+r /usr/local/cuda-xx.x/include/cu*.h

sudo chmod a+r /usr/local/cuda-xx.x/lib64/libcudnn*

验证

sudo cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

显存清理

sudo fuser -v /dev/nvidia* |awk '{for(i=1;i<=NF;i++)print "kill -9 " $i;}' | sudo sh