参考链接:Ubuntu实现多版本CUDA切换——CUDA9.0与CUDA10.0切换
(cudnn的安装是把一个.h头文件和几个lib放到cuda的对应目录下面)
将cuda-9.0切换成cuda-10.0的过程如下:
sudo rm -rf /usr/local/cuda #删除之前生成的软链接
sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda #生成新的软链接
完成之后即可看到nvcc -V输出的cuda版本已经变成了10.0。再想切回cuda-9.0只需要再用上述指令给9.0的路径生成新的同名软链接即可。
查看CUDA和Cudnn版本号:
cat /usr/local/cuda/version.txt
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2