Ubuntu 16.04: TensorFlow + CUDA8.0踩了无数坑的安装教程

这篇教程本身没有什么代码= =但是是踩过无数个坑还重装过Ubuntu之后 对各种教程的汇总
我用的是 Ubuntu 16.04 + NVIDIA GTX 960M,目测Ubuntu的系统安装过程差不多,但其他LInux发行版没有研究过。

一定要看官方教程文档!!!

TensorFLow的官网有一定概率需要自备梯

官方文档

  • Installing TensorFlow on Ubuntu
  • NVIDIA CUDA 安装文档

安装 CUDA 8.0

TensorFLow 目前没有支持 CUDA 9.0, 所以一定要装 8.0 ,直接按照 NVIDIA 官方教程下载安装 CUDA 9.0 就会踩到无数个坑……当然如果一定要选用最新版CUDA,可以参考 这篇教程

网上的很多教程认为 runfile 方式安装成功率会更高,但是我比较懒,喜欢一步到位的deb安装包
使用阿里云的deb源及阿里/google DNS(具体方法自行google),只需要执行
sudo apt-get install cuda-8-0
如果失败,那就只能按照官方教程一步步安装,但要记得下载 历史8.0版本

关于安装完成后添加环境变量:

官方文档里说只需在终端中运行上述两条export语句即可,但如果不将它们不写入/etc/profile文件的话,这样的环境变量在你退出终端后就消失了,不起作用了,所以写入才是永久的做法。

所以最好后续步骤按照 这篇教程 来进行
后续编译sample过程中如果出现错误,可以尝试用 make -k 替代 make

安装 cuDNN v6.0

不需要像官方教程那么复杂地再单独安装各种驱动,还是按照 这篇教程 中有关cuDNN的过程进行就好了。注意版本一定要是 v6.0 ,否则 TensorFlow 不支持。

安装 TensorFlow

这步很简单,直接按照官方文档中的步骤进行就可以。推荐用 virtualenv 方式安装。

你可能感兴趣的:(Ubuntu 16.04: TensorFlow + CUDA8.0踩了无数坑的安装教程)