0
点赞
收藏
分享

微信扫一扫

CUDA、cudnn安装教程

目录

  • ​​CUDA安装​​
  • ​​历史版本CUDA安装​​
  • ​​安装小测​​
  • ​​CUDNN安装​​
  • ​​选择系统对应的版本(我的环境切换为windows10了)​​
  • ​​cudnn安装版本查询​​

CUDA安装

资源链接:
​​​历史版本的安装​​​​当前最新的CUDA安装​​ 当前以ubuntu18.04的服务器环境,安装cuda10.02

历史版本CUDA安装

选择好对应版本之后,选择操作系统等一系列信息,你可以看见如下信息:
CUDA、cudnn安装教程_ubuntu

安装命令:

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. 第一步
    CUDA、cudnn安装教程_打开文件_02
  2. 第二步
    CUDA、cudnn安装教程_打开文件_03
  3. 第三步
    CUDA、cudnn安装教程_打开文件_04
  4. 第四步
    CUDA、cudnn安装教程_ubuntu_05
  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

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

清澈的爱,只为中国

举报

相关推荐

0 条评论