CUDA、cudnn安装教程


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安装

查找历史版本及其与CUDA的对应关系

选择系统对应的版本(我的环境切换为windows10了)

  1. 第一步
  2. 第二步
  3. 第三步
  4. 第四步
  5. 下载(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

安装的路径有可能比较复杂,可以根据自己的情况调整!


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM