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 )