Ubuntu18配置CUDA10.2 cudnn8.0.1(cuda_10.2.run格式+cudnn-10.2-linux-x64-v8.0.1.13.tgz格式)
1. 安装NVIDIA显卡驱动
如果使用的是GPU服务器,很大概率已经安装过了显卡驱动,使用如下命令检查驱动信息:
nvidia-smi
看见如下的打印信息,则说明已经有显卡驱动了:
如果没有显卡驱动,即输入nvidia-smi会报错,那么就需要安装驱动。
2. 安装cuda.run
注意,本文使用的是.run文件格式来安装CUDA,其他方式的还没有试。
首先从上面的显卡信息中知道我们需要的CUDA版本是10.2,那就去官网如下图:
如红色横线所示,很简单,只需要在你的ubuntu系统里面输入命令:
$ wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
进行下载
下载完毕后运行命令:
sudo sh cuda_10.2.89_440.33.01_linux.run
安装CUDA,安装步骤中具体选择如下所示:
(1)输入accept确认安装
(2)安装项选择,因为之前已经安装显卡驱动了,所以这块就不选择安装显卡驱动,上下键调节到“Install”,按enter开始安装
(3)设置环境变量
第一个
$ export PATH=/usr/local/cuda-10.2/bin:$PATH
第二个
$ export LD_LIBRARY_PATH=/usr/local/cuda10.2/lib64
(4)检查安装是否正确
$ nvcc -V
显示如下信息说明安装正确:
接下来测试案例是否能够正常运行。
(1) 编译并测试设备 deviceQuery:
$ cd /usr/local/cuda-10.2/samples/1_Utilities/deviceQuery
$ sudo make
$ ./deviceQuery
(2) 编译并测试带宽 bandwidthTest:
$ cd ../bandwidthTest
$ sudo make
$ ./bandwidthTest
如果这两个测试的最后结果都是Result = PASS,说明 CUDA 安装成功啦。
3 安装cudnn
(1)去官网,点击下图中连接开始下载
(如果使用的是SSH连接的服务器,可以在本地使用浏览器下载,然后使用rz上传到服务器中。如果是桌面型服务器,那就直接使用浏览器下载)
下载后的文件如下:
(2)在该级目录下,运行如下命令:
$ tar -zxvf cudnn-10.2-linux-x64-v8.0.1.13.tgz
(3)如上图解压出的cuda文件夹,在该级目录下,运行如下命令(注意路径是否一样):
$ sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
$ sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
(4)验证是否安装正确:
终端输入:
$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如下显示即安装成功