Win10+GeForce GTX 1060+Anaconda 快捷配置TensorFlow GPU环境

安装Anaconda

去 官网 下载即可，不赘述

更改Anaconda的源

打开powershell 输入命令

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

conda config --setshow_channel_urls yes

创建虚拟环境

打开powershell 输入命令

conda create -n tensorflow python=3.5

配置TensorFlow GPU版本

打开powershell 输入命令

conda install -c anaconda tensorflow-gpu 

• 这条命令会自动帮你下载CUDA、CUDNN等依赖，比自己配方便很多
• 注意这里还是从官方的源下的TensorFlow-GPU，可能会因为网络原因失败(之前换源只是为了安装其他包比较方便)，可以多试几次或者指定用清华的源

测试

启动任意一个使用了TensorFlow包的程序即可，运行时会显示你的显卡信息

或者随便测试一下矩阵乘法：

import tensorflow as tf
a = tf.constant([1.,2.,3.,4.,5.,6.], shape=[2,3], name='a')
b = tf.constant([1.,2.,3.,4.,5.,6.], shape=[3,2], name='b')
c = tf.matmul(a,b)

with tf.Session(config= tf.ConfigProto(log_device_placement=True)) as sess:
    print(sess.run(c))

代码引用自https://blog.csdn.net/linhai1028/article/details/79525507

总结

anaconda确实好用，省下来一天的配环境的时间
对了，如果运行时遇到什么dll库加载有问题的报错的话，可以去https://www.microsoft.com/zh-CN/download/details.aspx?id=53587 下载一个Visual C++ 2015 Redistributable，安装之后重启一下电脑，再试试
( 并不需要安装一个庞大的VS2015hhh )