导读

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

1. CUDA装置

  1. 返回官网下载合乎本人零碎版本的CUDA，依据本人电脑配置和零碎状况，抉择对应的选项，如下图：

  1. 依据网页的装置提醒操作，如下图：

  1. 关上终端进入到CUDA安装文件所在文件夹，赋予其可执行权力
sudo  chmod +x cuda_10.0.130_410.48_linux.run
  1. 装置
sudo ./cuda_10.0.130_410.48_linux.run

  1. 批改环境变量
sudo vi ~/.bashrcexport CUDA_HOME=/usr/local/cuda-10.0 # 依据版本批改export LD_LIBRARY_PATH=${CUDA_HOME}/lib64export PATH=${CUDA_HOME}/bin:${PATH}# 更新失效source ~/.bashrc
  1. 查看装置的版本信息
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.solitairetheme8mv cudnn-10.0-linux-x64-v7.4.2.24.solitairetheme8 cudnn
  1. 将解压后的文件复制到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*
  1. 查看版本 (可能不显示，但不影响应用)
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

本文由mdnice多平台公布