TensorFlow-gpu+win10+Anaconda+cuda8+cudnn5.1安装

1.安装前的准备
(1)下载VS2015,建议安装社区版,免费,安装时选择自定义,然后只安装visual studio C++,别的都不装,千万别装成全家桶
(2)下载DXSDK_Jun10.exe,安装时直接next就行,下载地址http://pan.baidu.com/s/1kUU4CCZ
(3)下载CUDA,地址http://pan.baidu.com/s/1i5A6e3b
(4)下载Anaconda,地址http://pan.baidu.com/s/1o8nvuTs
2.安装
安装CUDA和Anaconda请参考http://www.jianshu.com/p/c245d46d43f0,讲得很详细。
注:安装Anaconda时,我安装在默认目录下的,选的all users(要管理员权限),(不这样安总是出错)
3.安装Tensorflow
因为需要管理员权限,直接在Anaconda Prompt下安装不了,到命令提示符(管理员)下进行安装,用pip命令进行安装
首先更新pip,输入命令 python -m pip install –upgrade pip;
然后输入命令 pip install tensorflow-gpu,安装Tensorflow
安装时可以挂代理的全局模式,这样就不会安到一半停了。。。
4.测试
在Anaconda Prompt下输入python,然后输入下面代码进行测试
import tensorflow as tf
sess = tf.Session()
a = tf.constant(10)
b = tf.constant(22)
print(sess.run(a + b))
如果输出32,则安装成功

    原文作者:tensorflow
    原文地址: https://www.cnblogs.com/ctt0718/p/6800793.html
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞