先把大佬的博客贴在这里, 本文并不准备进行全面讲解,大佬的博客已经写得相当详细了,只对部分内容进行修改注释。
史上最全的ubuntu16.04安装nvidia驱动+cuda9.0+cuDnn7.0
一. 背景
这几天,准备入门知识图谱,可惜兴致勃勃地看完了很多入门教程后,被环境搭建这一关卡死,最后选择先跑通一个seq2seq的模型,暂时不打算管复杂的知识库搭建的操作。(看起来简单, 但是总是会报错,原谅我技术渣)。
由于demo需要gpu的运行环境,理论上,我可以直接修改成使用cpu运行的,不过,为了减少变数,还是乖乖选择了搭建gpu的环境,毕竟,这个我还是搭建过的,应该没有前面那个环境那么难。
然而,我想多了, 以前搭建的是windows的gpu环境,现在是要搭建linux的gpu环境,然后,遇到了很多坑,wc,为啥,做个东西这么难, 我滴?。
好了, 不说没有用的,接下来开始介绍流程。
二. 流程
这一部分并不是真正的教程,是一个根据以上博客进行的补充,再此膜拜大佬并恳请大佬的原谅。
1. 检查自己的gpu是否是cuda-capable
2. 下载安装包:NVIDIA驱动,cuda9.0,cudnn7。
其中,NVIDIA需要根据你系统和GPU的型号来下载相应的驱动,不然可能会导致安装失败,当时没多想,照着教程弄了很多次都没成功,才发现是因为下载的版本不适合我的GPU型号。
cuda9.0的下载需要到cuda下载页面,找到旧版本下载地址,然后下载安装。
cudnn7选择下载 cuDNN Library for Linux。
3. 安装NVIDIA驱动
3.1 禁用nouveau
我根据大佬的方法进行操作,都没有成功禁用nouveau,后来发现禁用后需要关机才能生效。
3.2 安装驱动
4. 安装CUDA-9.0
在操作中,没有办法在文本模式使用 gedit 命令,所以选择使用vim命令进行操作,如果没有vim,可以使用
sudo apt-get install vim
5. 安装cudnn7
如果下载的cudnn版本与博主的不同,比如我下载的是cudnn-v7.4.1.5,那么在建立软连接的时候就不能直接照搬大佬的命令,需要改为如下:
sudo chmod +r libcudnn.so.7.4.1 # 自己查看.so的版本
sudo ln -sf libcudnn.so.7.4.1 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig
这个需要根据实际来修改。
三. 总结
最后总算是成功配置好了环境,万岁,重装了好几次系统,还是很值得的。
四. 参考
1. 史上最全的ubuntu16.04安装nvidia驱动+cuda9.0+cuDnn7.0
2. windows 7 下cuda 9.0 卸载、cuda8.0 安装