折腾了两三天总算是装好了,感想有二:官方教程才是爸爸;有些教程可以让你少活二十年。
首先把官方教程膜拜上:官方教程
官方教程中给出的安装方法有两种:
Package Manager Installation
Runfile Installation
虽然本教程讲的是安装cuda9,但还是建议大家安装cuda8(在步骤2下载cuda8的deb安装包即可 CUDA8下载链接、cudnn),因为tensorflow1.4及之前版本对cuda9的支持还不是很好,望君且装且珍惜~
去tensorflow的GitHub看了一下,tensorflow已经到1.5版本了,在breaking changes 中明确说明了已经支持cuda9和cudnn7,所以大家可以放心的安装cuda9啦
目前网上大部分教程都是Runfile Installation安装的,然而这中方法很繁琐,需要关X Server,禁用nouveau等等,还会出现无法开机、无限循环登录等一些列毁天灭地的后果。如果侥幸顺利安装后,一个”CUDA driver version is insufficient for CUDA runtime version”的错误可以让你玩一年~
装了N次之后我已经有些绝望了,后来发现还有一种Package Manager Installation的方法,只需要四行命令就解决了,想起之前真的是天堂有路你不走,地狱无门你自投o(╥﹏╥)o
—————————————吐槽分割线——————————————————
1. Pre-installation Actions
这一步需要检查
- GPU是否支持CUDA
- Ubuntu版本是否受支持
- gcc版本检查
- Kernel Headers and Development Packages 是否已安装
这些步骤官方教程很详细,请移步官方教程第二节Pre-installation Actions
2. 下载Nividia Cuda Toolkit的deb安装包
CUDA Toolkit下载地址
根据自己的系统一步一步选择,注意主要下载的是.deb文件
3. 执行命令进行安装
在deb所在目录下依次执行以下命令即可。
1. sudo dpkg -i cuda-repo-ubuntu1604-9-1-local_9.1.85-1_amd64.deb
2. sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub
3. sudo apt-get update
4. sudo apt-get install cuda
以上命令其实就是选择完deb安装包后,官方给出的安装说明。
deb安装包会安装CUDA Toolkit和Driver Package,不需要自己安装驱动
安装时好像也会自动设置环境变量,记不太清了~如果没有的话请查看官方教程7.1.1. Environment Setup,设置一下环境变量。
4. 验证安装
重启系统然后进行验证
1. 检查GPU驱动版本和NVCC版本
2. 编译NVIDIA_CUDA-9.1_Samples下的例子,然后跑两个可执行文件。
官方文档也很清楚,请移步官方简称的7.2.3节Verify the Installation
——————————————胜利分割线—————————————————
如果你运气不是太差的话,应该已经顺利安装完成了。Congratulation!!!
如果不幸失败了,请在评论里留言,我会尽量帮助解决;也可以去FAQ看一下~ FAQ
好好学英语,抱官教大腿~
我们不生产教程,我们只是官方教程的搬运工(*╹▽╹*)