(1)下载需要的CUDA版本
https://developer.nvidia.com/cuda-toolkit-archive
(2)安装
sudo sh cuda_8.0.61_375.26_linux.run
(3)添加环境
gedit ~/.bashrc
在文件末尾添加:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64
export PATH=$PATH:/usr/local/cuda-8.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-8.0
(4)使文件有效
source ~/.bashrc
(5)查看安装是否成功
cat /usr/local/cuda-8.0/version.txt
或者
nvcc -V ## nvcc -version
(1)下载CUDNN
https://developer.nvidia.com/cudnn
需要注册用户,然后登录下载。注意cudnn的版本要和cuda的版本对应。
(2)安装
sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64/
sudo chmod a+r /usr/local/cuda-8.0/include/cudnn.h
sudo chmod a+r /usr/local/cuda-8.0/lib64/libcudnn*
(3)查看是否安装成功
cat /usr/local/cuda-8.0/include/cudnn.h | grep CUDNN_MAJOR -A 2
以上步骤1和步骤2就是但版本CUDA的安装
例如:将CUDA-8.0更换为CUDA-10.0。这里,CUDA-8.0和CUDA-10.0以正确安装完成。可以通过创建cuda软链接指向不同版本的cuda版本。
首先,修改CUDA的路径,
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
这里cuda软链接指向cuda-8.0版本。
然后,删除指向cuda-8.0的软链接cuda,重新创建一个指向cuda-10.0版本的软链接。
rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda
nvcc --version