美文网首页
win10 安装tensorflow GPU版本

win10 安装tensorflow GPU版本

作者: 酱紫冻 | 来源:发表于2018-08-12 10:28 被阅读0次

    安装cuda

    官网下载cuda,本文安装的是9.0版本。

    下载cudnn

    下载对应cuda9.0的cudnn,本文下载的是cudnn7.1,下载解压,将解压到的文件夹里面的bin,lib,include三个文件夹的文件复制到cuda对应的文件夹里。

    安装Anaconda3

    TensorFlow 只支持Python3以上版本,因此需要安装Anaconda3.

    安装 TensorFlow

    1. 首先,为TensorFlow创建一个虚拟环境:
    conda create --name tensorflow-gpu python=3.5
    

    如果报LOCKERROR:It looks like conda is already doing something.执行

    conda clean --lock
    

    2)激活/进入虚拟环境

    activate tensorflow-gpu
    

    3)安装TensorFlow with GPU support

    pip install  --upgrade tensorflow-gpu
    

    测试TensorFlow是否安装完成

    import tensorflow as tf
    hello =  tf.constant('Hello, TensorFlow!')
    sess = tf.Session()
    print(sess.run(hello))
    #输出:Hello, TensorFlow!
    a = tf.constant(10)
    b = tf.constant(20)
    print(sess.run(a+b))
    #输出:30
    

    相关文章

      网友评论

          本文标题:win10 安装tensorflow GPU版本

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