进入官网,找到对应的10.2版本,选择要安装的平台;
在终端执行命令:
wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
继续执行命令:
sudo sh cuda_10.2.89_440.33.01_linux.run
输入accept
注意:如果你已经安装了driver,这里需要去掉,否则会报错!!!
点击安装:
稍等一会:
输入nvcc -V查看
发现还是10.0版本,因此需要修改一下环境变量:
sudo gedit ~/.bashrc
并注释掉之前的变量,加入下面变量。如果使用之前的版本,可以注释掉新安装的变量。
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-10.2/bin:$PATH
export CUDA_HOME=usr/local/cuda-10.2$CUDA_HOME
加入新变量:
保存之后,为了使这个修改立即生效,还需要输入如下代码(如果不执行 source 命令,则需重启系统才能生效):
source ~/.bashrc
这样就变成10.2啦!可以安装多个版本的cuda,使用时切换环境变量即可!!!
cuDNN下载安装:https://developer.nvidia.com/rdp/cudnn-download
需要登录下载:
选则第一个:
通过命令解压文件:
tar -xzvf cudnn-10.2-linux-x64-v8.0.4.30.tgz
并进入到相应目录,运行以下命令:
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.2/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.2/lib64
sudo chmod a+r /usr/local/cuda-10.2/include/cudnn.h
sudo chmod a+r /usr/local/cuda-10.2/lib64/libcudnn*
使用查看安装版本:
dpkg -l | grep cudnn
发现问题,版本不对,还是之前版本。
在cudNN官网上下载:
使用命令安装:
sudo dpkg -i
再次查看版本:
安装Pytorch:
测试: