CUDA安装
资源链接:
历史版本的安装
当前最新的CUDA安装
当前以ubuntu18.04的服务器环境,安装cuda10.02
历史版本CUDA安装
选择好对应版本之后,选择操作系统等一系列信息,你可以看见如下信息:
安装命令:
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
如图所示,根据命令在命令行执行就可以了
安装小测
在终端执行下面的命令进行cuda测试:
nvidia-smi
CUDNN安装
选择系统对应的版本(我的环境切换为windows10了)
- 第一步
- 第二步
- 第三步
- 第四步
- 下载(windows10对应版本)
cuda10.2对应的cudnn7.6.5.23:
https://developer.download.nvidia.cn/compute/machine-learning/cudnn/secure/7.6.5.32/Production/10.2_20191118/cudnn-10.2-windows10-x64-v7.6.5.32.zip?lSdULl8SFl_ABsp7SYDWfquUYycvAs1BC5IKjQckJJcf7RuyDq7lK1anjgmsAvgtkibFHNgBCdf-W4-Q0g7lleaLBTywu4xHrSAXBfrMaghufQFFdYhrWk6Cf6SQo8zJa0ydTBX91oAMXQnjIIhQGlhjoAkuTPDlPn3TkZv3A3q_Y29i1yopf4nIOe9TPUrXou2FpqQqcdV8EVcHU5XUxh9xl7CF64yDECsorkI
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
若方法一、方法二都失效了,你还安装了cudnn?OK,进入/usr/local/cuda/include/
下查看上面两个文件,打开文件如果内容明显很少,可能是解压的时候出现问题!
方法三
$ cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2
安装的路径有可能比较复杂,可以根据自己的情况调整!