美文网首页
Tensorflow、CUDA、CUDNN 、python 版本

Tensorflow、CUDA、CUDNN 、python 版本

作者: Young1217 | 来源:发表于2019-07-16 11:46 被阅读0次

    原文:
    Tensorflow不同版本要求与CUDA及CUDNN版本对应关系
    tensorflow各个版本的CUDA以及Cudnn版本对应关系
    Ubuntu16.04 + CUDA9.0 + cuDNN7.3 + Tensorflow-gpu-1.12 + Jupyter Notebook 深度学习环境配置

    Linux端:https://tensorflow.google.cn/install/source

    图片.png
    图片.png 图片.png
    图片.png
    图片.png
    图片.png
    图片.png 图片.png

    安装tensorflow-gpu版本
    直接使用sudo pip install tensorflow-gpu 命令安装(前提已经安装pip 或pip3),这时会发现安装的是最新版本,有可能与cuda版本不匹配。import tensorflow也会报错。可以下载一个gpu版本的tensorflow,离线在本地使用pip安装。(下载地址:http://pypi.org/project/tensorflow-gpu/1.12.0/#files)

    sudo pip install ******(下载的文件)
    

    安装tensorflow-gpu版本(如果能上外网)

    pip install tensorflow-gpu==1.10
    

    相关文章

      网友评论

          本文标题:Tensorflow、CUDA、CUDNN 、python 版本

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