Windows10下通过Anaconda安装TensorFlow-gpu版

安装前准备:

TensorFlow 有两个版本: CPU 版本和 GPU 版本。GPU 版本需要 CUDAcuDNN 的支持,CPU 版本不需要。如果你要安装 GPU 版本,请先确认你的显卡支持 CUDA。我安装的是 GPU 版本TensorFlow1.3。

官网下载Anaconda;(我下的版本默认python3.6,后面要降到3.5,也可以直接下载默认python3.5版的Anaconda3)


官网下载CUDA:cuda_8.0.61_win10;下载时选择 exe(local)




cuDNN:cudnn-8.0-windows10-x64-v6.0;(其实是一个压缩包)

开始安装:

1.安装CUDA

直接双击下载的exe文件就可以了,就像安装正常的其他软件一样,安装时间有点长。安装完之后系统变量会自动为你添加上,这个不用管。

测试一下是否安装成功,命令行输入 nvcc -V ,看到版本信息就表示安装成功了。

2.安装cuDNN库

把下载好压缩包,解压放到任何一个目录下就行,然后把其中的bin目录路径添加到Path环境变量里。我 就直接放到CUDA的安装目录下。

                 

3.安装Anaconda

就跟安装普通的软件一样,全部选择默认即可,注意勾选将python3.6添加进环境变量。

安装好了之后,我们可以通过命令来查看Anaconda已经安装了哪些包。

开始菜单->Anaconda3—>Anaconda Prompt :

输入:conda list

修改默认的python版本:

我下的Anaconda默认python是3.6,但是听说tensorflow对3.5较为友善,所以创建了个3.5的环境。

1.创建新环境

    打开Anaconda Prompt,输入一下命令:

    conda create -n py35 python=3.5 anaconda

新环境创建成功后可以在Anaconda安装目录中的envs文件夹中查到。

2.激活新环境

输入:activate  py35

3.确认环境

运行:conda info --envs

显示环境中的所有列表。

4.安装TensorFlow

虽然可以通过pip3 install --upgrade tensorflow-gpu来直接安装,但安装的版本不一定是想要的版本。

这里用清华镜像源安装。

1.打开Anaconda Prompt.添加清华的镜像源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --set show_channel_urls yes

2.在当前用户目录下(C:\Users\xxxx)下新建pip目录,在pip目录下新建pip.ini文件,配置文件:

[global]                                                                     
index-url = http://mirrors.aliyun.com/pypi/simple/
//https://pypi.tuna.tsinghua.edu.cn/simple 清华源  
[install]                                                                      
trusted-host=mirrors.aliyun.com                             


3.建立TensorFlow环境

conda create -n tensorflow Python=3.5

4.激活TensorFlow环境

activate TensorFlow

5.安装TensorFlow

pip install --upgrade --ignore-installed tensorflow-gpu

 安装TensorFlow指定版本(清华源上有的,更换链接最后的版本名称就行了)
pip install --upgrade https://mirrors.tuna.tsinghua.edu.cn/tensorflow/windows/gpu/tensorflow_gpu-1.3.0rc0-cp35-cp35m-win_amd64.whl                                                                                                                                                                                                        

安装完成

6.测试

在TensorFlow环境里打开Python:

import tensorflow as tf                           
                                                                
hello = tf.constant("Hello!TensorFlow")

sess = tf.Session()                                   

print(sess.run(hello))                               




出现b'Hello!TensorFlow'表示TensorFlow已成功装好。大笑

  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 5
    评论
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值