系统: ubuntu 18.04
Driver Version: 430.50
CUDA Version: 10.1
一、安装N卡驱动
方法一:通过ubuntu中的“软件更新“升级原有驱动(简单无脑)
https://jingyan.baidu.com/article/d7130635c5a86113fdf47532.html
方法二:通过命令行安装(小白绕行)
https://blog.csdn.net/u013832707/article/details/93157805
二、安装CUDA
2.1 下载cuda
下载地址: https://developer.nvidia.com/rdp/cudnn-download
选择runfile格式的CUDA文件下载
2.2 安装cuda
(1)运行如下指令开始安装:
sudo sh cuda_10.1.105_418.39_linux.run
(2)输入"accept" 如下图:
(3)选择“install” (注意:取消驱动安装,如下图)
2.3 配置环境变量
打开 bashrc
sudo vim ~/.bashrc
在文件最后一行添加:
export PATH=/usr/local/cuda-10.1/bin:/usr/local/cuda-10.1/NsightCompute-2019.1${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
刚才的配置生效
source ~/.bashrc
三、安装cuDNN
3.1下载 cuDNN
下载地址:https://developer.nvidia.com/rdp/cudnn-download
选择 "cuDNN Library for Linux“,如下图:
3.2配置cuDNN环境:
解压下载的文件,并进入到相应目录,运行以下命令:
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h
sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*
参考: