步骤1:
单独安装显卡驱动
步骤2:
安装多个版本的cuda,以及对应的cudnn(记住,cudnn安装到/usr/local/cuda-版本号,而不是usr/local/cuda/中)。
环境变量:
export CUDA_HOME=/usr/local/cuda
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
步骤3:
切换cuda
cd /usr/local
sudo rm -rf cuda
sudo ln -s /usr/local/cuda-版本号 /usr/local/cuda #添加软连接
查看cuda
nvcc -V
查看cudnn
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