本文将介绍如何在Ubuntu
下安装深度学习需要使用的CUDA
和 cudnn
。
CUDA
,根据自己电脑配置和系统情况,选择对应的选项,如下图:CUDA
安装提示
CUDA
安装文件所在文件夹,赋予其可执行权利sudo chmod +x cuda_10.0.130_410.48_linux.run
sudo ./cuda_10.0.130_410.48_linux.run
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
nvcc -V
# cat /usr/local/cuda/version.txt
nvcc信息
CUDA
对应的cudnn
版本(要注册账号才能下载)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
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*
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