美文网首页
tensorflow+cuda+cudnn 安装最佳实践

tensorflow+cuda+cudnn 安装最佳实践

作者: 被代码捆住腿的摄影师 | 来源:发表于2018-10-20 22:58 被阅读0次

    优先阅读官方文档!!!

    • 使用环境:ubuntu 16.04 + Nvidia 1060

    安装tensorflow+cuda+cudnn的正确方法:

    1. 确定cuda和cudnn的版本
      阅读tensorflow的官方文档,找到下面这张图:
    经过测试的tf,cuda,cudnn的版本
    该图片可以在此处找到,从中确定经过测试的兼容的cuda和cudnn的版本。
    1. 安装cuda
      不需要单独安装nvidia的显卡驱动,因为cuda里已经包含了合适版本的显卡驱动。
      Nvidia的网站下载cuda,需要下载以前的版本去Legacy Releases里下载。详细的安装方法照着里面的文档进行。推荐使用deb(local)封装安装。
    2. 安装cudnn
      和前面一样在官网下载,并参照官方文档安装。也推荐使用deb格式安装。
    3. 安装tensorflow
      推荐使用pip安装
      pip3 install tensorflow-gpu

    完成!

    相关文章

      网友评论

          本文标题:tensorflow+cuda+cudnn 安装最佳实践

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