您的位置:首页 > 其它

ubuntu16.04安装cuda9——简明教程

2017-12-19 23:13 253 查看
折腾了两三天总算是装好了,感想有二:官方教程才是爸爸;有些教程可以让你少活二十年。

首先把官方教程膜拜上:官方教程

官方教程中给出的安装方法有两种:

Package Manager Installation

Runfile Installation

虽然本教程讲的是安装cuda9,但还是建议大家安装cuda8(在步骤2下载cuda8的deb安装包即可 CUDA8下载链接),因为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

好好学英语,抱官教大腿~

我们不生产教程,我们只是官方教程的搬运工(*╹▽╹*)
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  ubuntu cuda 安装教程