美文网首页我爱编程
Ubuntu 16.04: TensorFlow + CUDA8

Ubuntu 16.04: TensorFlow + CUDA8

作者: lja672 | 来源:发表于2017-11-04 21:58 被阅读0次

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

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

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

    官方文档

    安装 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

        本文链接:https://www.haomeiwen.com/subject/vbyqmxtx.html