编辑bashrc文件
sudo gedit ~/.bashrc
source ~/.bashrc
第一种方法:把路径中的两个cuda-8.0 替换为cuda-10.0 然后source
第二种方法: 都改为cuda 然后source
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
注:
CUDA 默认安装路径是 /usr/local/cuda-8.0、 /usr/local/cuda-9.0、/usr/local/cuda-10.0。
/usr/local/cuda 是一个符号链接文件,指向 /usr/local/cuda-8.0、 /usr/local/cuda-9.0、/usr/local/cuda-10.0 文件夹。使用 ls -l 指令可以查看文件的链接关系
切换至 cuda-8.0
sudo rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-8.0 /usr/local/cuda
nvcc --version
which nvcc
切换至 cuda-10
.0
sudo rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda
nvcc --version
which nvcc
查看cudnn的版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2