当前位置:   article > 正文

(二)Cuda10.2与Cudnn7.6.5安装

cudnn7.6.5

Cuda官网所有版本

Cudnn所有版本

这里以Cuda10.2和Cuddn7.6.5为例

Cuda10.2安装

下载地址

runfile安装

wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.runsudo 
sh cuda_10.2.89_440.33.01_linux.run
  • 1
  • 2

这里选择不安装驱动

# 如何检查驱动是否安装
sudo lshw -c video
# 如果在configuration这一行中有driver=nvidia的字符串,那么就说明nvidia的驱动安装ok
  • 1
  • 2
  • 3

设置环境变量

export PATH=/usr/local/cuda-10.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda10.2/lib64 
  • 1
  • 2

重启电脑后在终端输入:env,检查环境变量中有无刚加入的变量。

安装后检查

# 会输出CUDA的版本信息。
nvcc -V
  • 1
  • 2

Cuddn7.6.5安装

# 下载
wget https://developer.nvidia.com/compute/machine-learning/cudnn/secure/7.6.5.32/Production/10.2_20191118/cudnn-10.2-linux-x64-v7.6.5.32.tgz
# 解压
tar -xzvf cudnn-10.2-linux-x64-v7.6.5.32.tgz
  • 1
  • 2
  • 3
  • 4

复制文件

# 终端输入以下命令将文件复制到CUDA中
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
  • 1
  • 2
  • 3
  • 4

复制后即完成cuDNN安装

验证是否成功

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
  • 1

如下显示即安装成功

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小丑西瓜9/article/detail/128567
推荐阅读
相关标签
  

闽ICP备14008679号