CUDA下载安装
下载地址,选择适合自己系统的版本
我这里是CUDA8.0,百度网盘下载地址,密码:9yf5
依次执行下列命令
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
至此,CUDA安装完成。
修改环境变量
使用下面命令打开bashrc文件
sudo gedit ~/.bashrc
在文件最后加上cuda配置
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
然后保存。
source ~/.bashrc
验证
nvcc -V
(watch -n 1) nvidia-smi
CUDA成功安装。
cuDNN配置
下载链接,同样是选择合适的版本
我的对应版本cudnn6.0,百度网盘下载地址,密码:3hov
解压,执行以下命令
tar -xzvf cudnn-8.0-linux-x64-v6.0.tgz
得到一个cuda文件夹,里面有两个文件夹,分别是include和lib64
执行以下命令,完成文件复制
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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
cuDNN配置完成。