美文网首页
在win10下安装Tensorflow1.3-gpu(CUDA8

在win10下安装Tensorflow1.3-gpu(CUDA8

作者: 给力哥 | 来源:发表于2017-09-06 15:03 被阅读0次

    前提:安装的是GPU版本的tensorflow,CPU的直接安装就行了,不需要第一第二步

    写这篇文章的时候,tensorflow最新版已经是1.3,网上的安装教程都是1.3之前的,cuDNN也都还是用的5.1,所以踩了坑,那就记录下来,让后面的朋友少踩点

    1.下载安装CUDA8.0
    https://developer.nvidia.com/cuda-80-ga2-download-archive
    2.下载安装cuDNN6.0,这里不要下5.1了,7.0我自己尝试也是不行的,下载需要登录,耐心点吧
    https://developer.nvidia.com/rdp/cudnn-download
    安装完需要设置系统环境变量

    image.png

    3.安装tensorflow-gpu
    1.3版本已经支持python3.6了
    只有python3的用
    pip install --upgrade tensorflow-gpu
    双版本python的用
    pip3 install --upgrade tensorflow-gpu

    最后打开cmd

    image.png

    理论上没报错就代表完成了

    相关文章

      网友评论

          本文标题:在win10下安装Tensorflow1.3-gpu(CUDA8

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