关于深度学习:深度学习之CUDA-cudnn-安装

导读

本文将介绍如何在 Ubuntu 下装置深度学习须要应用的CUDA 和 cudnn。

1. CUDA 装置

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. 官网下载与 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

本文由 mdnice 多平台公布