Ubuntu16.04 安装CUDA和CUDNN
1. 安装NVIDIA驱动
卸载掉原有驱动(如果是新配机器可以不用)
sudo apt-get remove –purge nvidia
安装驱动有两种办法
- 用ubuntu默认的附加驱动安装
- 在nvidia官网下载驱动安装
最近发现新官网下载的驱动有时问题比较多, 直接使用附加驱动nvidia-384就可以
系统默认驱动安装
进入附加驱动->选择nvidia-384->应用更改
安装成功后重启,检查nvidia-smi,有打印gpu信息的表格就说明成功了
安装cuda
- 下载
到官网下载, 首选runfile
需要注意驱动版本, 我的版本对应的是cuda9.0 - 安装
安装前需要先关闭图形界面
sudo service lightdm stop
安装runfile文件
注意: 安装时让选择是否安装驱动, 选择no
安装完成后打开图形界面chmod +x *.run sudo ./cuda_9.0.176_384.81_linux.run
sudo service lightdm restart
安装cudnn
- 下载
到官网下载, 需要注册 - 安装
解压cudnn-9.0-linux-x64-v7.tgz,
tar -zxvf cudnn-9.0-linux-x64-v7.0.tgz
复制解压好的文件到/usr/local/cuda下sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
添加环境变量
增加环境变量到~/.bashrc中
export CUDA_HOME=/usr/local/cuda
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
一切就绪重启计算机