我的驱动版本为450.102.04,这里选择cuda10.0+cudnn7.5
首先安装驱动
查看驱动方式nvidia-smi
下载cuda10.0
https://developer.nvidia.com/cuda-toolkit-archive
下载cudnn7.5
https://developer.nvidia.com/rdp/cudnn-archive
这里提供百度网盘链接
链接: https://pan.baidu.com/s/17wjgmWMCcKsr32Fc063-3g 密码: t5ws
下面安装cuda
cuda_10.0.130_410.48_linux.run和cuda_10.0.130.1_linux.run
在这两个文件所在文件夹下打开终端,执行安装命令
sudo sh cuda_10.0.130_410.48_linux.run
sudo sh cuda_10.0.130.1_linux.run
一个一个来安装
第一个问是否要装驱动,选择no,其余的都选yes。
配置环境变量,打开.bashrc
在主目录下ctrl+h,打开.bashrc文件,或者gedit ~/.bashrc
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存后进行source
source ~/.bashrc
安装完毕,nvcc --version或者cat /usr/local/cuda/version.txt检查是否安装成功
下面安装cudnn
按下面顺序安装
sudo dpkg -i libcudnn7_7.5.0.56-1+cuda10.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.5.0.56-1+cuda10.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.5.0.56-1+cuda10.0_amd64.deb
将第二个中的cudnn.h拷贝到/usr/local/cuda/include文件夹中,要sudo才可以复制
sudo cp /usr/include/cudnn.h /usr/local/cuda/include
sudo chmod a+x /usr/local/cuda/include/cudnn.h
安装完毕,cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2检查是否安装成功