深度学习之CUDA + cudnn 安装

导读

本文将介绍如何在Ubuntu下安装深度学习需要使用的CUDA 和 cudnn。

1. CUDA安装

1. 前往官网^[1]下载符合自己系统版本的CUDA，根据自己电脑配置和系统情况，选择对应的选项，如下图：

2. 根据网页的安装提示操作，如下图：

3. 打开终端进入到CUDA安装文件所在文件夹，赋予其可执行权利

sudo  chmod +x cuda_10.0.130_410.48_linux.run

4. 安装

sudo ./cuda_10.0.130_410.48_linux.run

5. 修改环境变量

sudo vi ~/.bashrc

export CUDA_HOME=/usr/local/cuda-10.0 # 根据版本修改
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}

# 更新生效
source ~/.bashrc

6. 查看安装的版本信息

nvcc -V
# cat /usr/local/cuda/version.txt

2. cudnn安装

1. 官网^[2]下载与CUDA对应的cudnn版本（要注册账号才能下载）

2. 打开终端进入cudnn下载目录，对其进行解压

tar -zxvf cudnn-10.0-linux-x64-v7.4.2.24.solitairetheme8
mv cudnn-10.0-linux-x64-v7.4.2.24.solitairetheme8 cudnn

3. 将解压后的文件复制到cuda目录下

sudo cp cudnn/include/cudnn*.h /usr/local/cuda/include 
sudo cp -P cudnn/lib/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

参考资料

[1]

CUDA官网: https://developer.nvidia.com/cuda-downloads

[2]

cudnn: https://developer.nvidia.com/cudnn