安装cuda
1.到官网下载安装包,(https://developer.nvidia.com/cuda-toolkit-archive)下载CUDA
建议下载:cuda_9.0.176_384.81_linux.run
2.到下载到本地的路径,开个命令窗口:
sudo chmod 777 cuda_9.0.176_384.81_linux.run #给用户增加读写和执行的权限
sudo ./cuda_9.0.176_384.81_linux.run
注意:执行上述命令约1分钟后会出现 0%信息,此时长按回车键(空格键)让此百分比增长,直到100%,然后按照提示操作即可,先输入 accept ,然后让选择是否安装 nvidia 驱动,这里因为已经安装过显卡驱动就不能再重复安装,所以输入“n”不安装。剩下的选择则都输入“y”确认安装或确认默认路径安装,开始安装。
3.
打开~/.bashrc文件:
sudo gedit ~/.bashrc
将以下内容写入到~/.bashrc尾部:
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
使该配置生效:一定要执行啊 ,如果同时打开多个终端,执行过后,重新打开,否则后面的错误也是不断的.
source ~/.bashrc
在/etc/profile文件中添加CUDA环境变量:
sudo gedit /etc/profile
打开文档都在文档结尾加上下面两句:
PATH=/usr/local/cuda/bin:$PATH
export PATH
保存后, 执行下列命令, 使环境变量立即生效:
source /etc/profile
同时添加lib库路径,在 /etc/ld.so.conf.d/新建文件 cuda.conf
sudo gedit /etc/ld.so.conf.d/cuda.conf
在文中加入下面内容:
sudo ldconfig
执行以下命令测试cuda是否安装成功
cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
sudo make
sudo ./deviceQuery
可参考博客:
Ubuntu16.04上配置CUDA 8.0详细教程
安装cudnn
上官网下载对应的cudnn https://developer.nvidia.com/cudnn
建议下载:cudnn-9.0-linux-x64-v7.3.0.29.solitairetheme8
1.在下载的到本地文件的路径,打开命令行
tar zxvf cudnn-8.0-linux-x64-v5.1.tgz #解压文件
cd进入解压之后的include目录,在命令行进行如下操作:
sudo cp cudnn.h /usr/local/cuda/include/ #复制头文件
2.
再cd进入lib64目录下的动态文件进行复制和链接:(7.3.0为对应版本具体可修改)
sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.7 #删除原有动态文件
sudo ln -s libcudnn.so.7.3.0 libcudnn.so.7 #生成软衔接
sudo ln -s libcudnn.so.7 libcudnn.so #生成软链接
可参考博客:cudnn 安装步骤 - alexanderkun - 博客园
安装完后,检测下是否都安装成功:
检查cuda:
cat /usr/local/cuda/version.txt
检查cudnn
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2