1.Ubuntu16.04安装CUDA10.2
具体参考:https://blog.csdn.net/qq_41251963/article/details/109401989 (如果侵权,速删)
环境:Ubuntu 16.04(已安装驱动)
安装: CUDA 10.2 & cuDNN 7.6.5
在终端输入:
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,这里要去掉,否则会报错!
然后选择 Install
查看CUDA版本号:
nvcc -V
如果提示:程序“nvcc”尚未安装。 您可以使用以下命令安装:sudo apt install nvidia-cuda-toolkit
不要这样安装!
正确的解决方案:
vim ~/.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
至此CUDA10.2就安装好了~
2.安装cudnn
cuDNN下载链接:https://developer.nvidia.com/rdp/cudnn-archive
下载 7.6.5版本!!!千万别下载8.0.4版本,巨坑!8.0.4版本不知道啥问题,就安装不上,鬼知道我安装了几遍...乌鱼子...
选择第三行的cuDNN Library for Linux下载
解压至当前文件夹
tar -xzvf cudnn-10.2-linux-x64-v7.6.5.32.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*
查看版本号看是否安装成功:
cat /usr/local/cuda-10.2/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果没有反应,可以重启一下,或者可以试试输入:
//重启
sudo reboot
//cuDNN8.0之后将版本号放入另一个头文件中
cat /usr/local/cuda-10.2/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
至此cuDNN7.6.5就安装好了~
3安装Pytorch1.6
CUDA 10.2与Pytorch1.6匹配
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch