ubuntu16.04安装cuda9.0和cudnn对应版本教程
为了学习过程中的环境需求,所以笔者需要搭建对应的学习环境。由于笔者接触ubuntu时间不长,所以在配置过程踩了很多的坑。在这里笔者把自己的经验给大家分享一下。
本次教程是在ubuntu16.04系统和拥有nvidia显卡驱动的基础上进行的。笔者驱动版本450.57,能够满足cuda9.0需要。
1.CUDA安装
1)CUDA9.0下载
CUDA系列的包可以直接去NVIDIA官网进行下载,而且下载速度也比较快。读者可以根据自己电脑驱动版本去下载对应的版本。选择下载类型时,推荐使用runflie类型的文件,安装起来简单而且成功率较高。如图所示
下载完毕后进入下载的目录终端,为了安装方便我将下载后的文件命名为cuda.run(不修改文件名也可直接输入原始名字),输入命令检测,序号不同需要重新下载。
md5sum cuda.run
2)安装cuda
(1)禁用nouveau
如果实现安装过驱动,那么应该禁用过nouoveay,可以使用命令进行查看:
lsmod | grep nouveau
如果没有输出,那么禁用成功,如果没有禁用过可在终端输入命令:
sudo gedit /etc/modprobe.d/blacklist.conf
然后会打开一个文档,再在文档的最后添加代码:
blacklist nouveau
options nouveau modeset=0
主要目的是为了防止驱动发生冲突。
保存后退出,然后在终端输入命令,使修改的内容生效,等待重启:
sudo update-initramfs -u
reboot
(2)在字符界面进行安装
- 紧接上一步的重启电脑过后,可以直接使用Ctrl+Alt+F1进入字符界面,输入自己的帐号密码进行登录。登录成功后会出现终端一样的命令输入字符。
- 关闭图形界面:
sudo service lightdm stop
- 执行安装命令
sudo sh cuda.run
会显示一大串的信息,主要是用户许可证的信息,按住空格键,最下方显示的进度会到100%
接着会出现进行选择的命令,输入内容为:
accept -> n -> y -> y ->y (对于没有选项的命令输入回车)
重启图形界面:
sudo service lightdm start
如果没有返回图形界面,使用快捷键 Ctrl+Alt+F7可返回图形界面。
- 设置环境变量
打开终端输入命令:
sudo gedit ~/.bashrc
在打开的文档的最后方加入下列语句:
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
建议加上注释,方便以后添加和删除环境路径。
(3)验证驱动版本和cuda版本
cat /proc/driver/nvidia/version
nvcc -V
(4)编译例程
- 编译例程文件:
打开终端输入命令:
cd NVIDIA_CUDA-9.0_Samples
make
编译时间可能较长,请耐心等待。
- 运行例程文件:
进入~/NVIDIA_CUDA-9.0_Samples/bin/x86_64/linux/release,可以使用命令进入也可通过文件夹进入在打开终端,输入命令:
./deviceQuery
得到Result = PASS如图。
- 检查系统和CUDA-Capable device的连接情况,终端输入 :
./bandwidthTest
得到输出结果如下图所示,表示安装成功。
3)安装cudnn
(1)cudnn文件下载
在NVIDIA官网下载:cudnn下载
选择最新版本进行下载,但是要注意匹配自己的cuda版本,选择第三个文件,cuDNN Library for linux。
下载结束后会获得一.solitairetheme8的文件,我们需要将文件转换为.tgz类型的文件,方便提取。进入下载目录打终端输入命令进行
cp cudnn-9.0-linux-x64-v7.6.3.30.solitairetheme8 cudnn-9.0-linux-x64-v7.6.3.30.tgz
tar -xvf cudnn-9.0-linux-x64-v7.6.3.30.tgz
然后在提取的目录下打开终端输入命令:
sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64/
(2)查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
此时我们就已经安装成功了。
参考文献:
[1]: https://blog.csdn.net/zbr794866300/article/details/106574704
[2]: https://blog.csdn.net/Super_jm_/article/details/78397074?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-8.nonecase&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-8.nonecase