Ubuntu安装CUDA和cuDNN

我的环境

Ubuntu16.04_x64

CUDA_9.2

uDNN_7.1.4

一、安装CUDA

1.进入官网，根据自己的系统选择版本开始下载

链接

2.下载完成后进入目录，按照官网给定的四条命令进行安装

sudo dpkg -i cuda-repo-ubuntu1604-9-2-local_9.2.148-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-2-local/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

如果存在依赖项缺失可以使用命令修复

apt-get -f install

3.安装完成后进行环境变量的配置

export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

注意其中的版本号要和自己下载的对应。

二、安装cuDNN

1.这里一定要根据上一步选择的CUDA版本选择对应的cuDNN的版本。下载前应该需要注册一个NVIDIA账号

链接

2.进入下载目录，进行解压，得到cuda文件夹

tar zxvf cudnn-9.2-linux-x64-v7.1.tgz

3.将cuda文件夹中的文件拷贝到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*

4.可以用如下命令检测是否安装成功

babao@babao:~$ nvidia-smi
Mon Aug  6 13:17:44 2018       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 396.37                 Driver Version: 396.37                    |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 960M    Off  | 00000000:01:00.0 Off |                  N/A |
| N/A   41C    P5    N/A /  N/A |    264MiB /  2004MiB |     19%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0       991      G   /usr/lib/xorg/Xorg                           155MiB |
|    0      1876      G   compiz                                       107MiB |
+-----------------------------------------------------------------------------+

大功告成