赞
踩
这里以Cuda10.2和Cuddn7.6.5为例
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
这里选择不安装驱动
# 如何检查驱动是否安装
sudo lshw -c video
# 如果在configuration这一行中有driver=nvidia的字符串,那么就说明nvidia的驱动安装ok
export PATH=/usr/local/cuda-10.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda10.2/lib64
重启电脑后在终端输入:env,检查环境变量中有无刚加入的变量。
安装后检查
# 会输出CUDA的版本信息。
nvcc -V
# 下载
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
# 终端输入以下命令将文件复制到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*
复制后即完成cuDNN安装
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如下显示即安装成功
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。