Windows8.1 安装tensorflow gpu版本
参考:https://zhuanlan.zhihu.com/p/32284273
http://blog.csdn.net/akon_wang_hkbu/article/details/78478513
安装CUDA和cuDNN
1. CUDA 8.0
下载地址:https://developer.nvidia.com/cuda-downloads
2.cuDNN v6.0
下载地址:https://developer.nvidia.com/rdp/cudnn-download
也可以到我的百度云下载:链接:https://pan.baidu.com/s/1gftmqR5 密码:zinu
解压后对应部分覆盖到C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v8.0目录即可(include、bin、lib中内容)。并进行相应配置
3. 安装TensorflowGPU 1.4
由于Anaconda 可以提供完整的科学计算库,所以直接使用Anaconda 来进行相关的额安装。
4.安装Anaconda
下载地址:https://www.anaconda.com/download/
这里我们下载Python 3.6 64bit 的Anaconda 5,直接安装即可。
5. 在Anaconda中安装TensorFlow GPU 1.4
conda create --name tf python=3.6 #创建tf环境
activate tf #激活tf环境
pip install --ignore-installed --upgrade tensorflow-gpu
备
其他命令:
deactivate tf #退出tf环境
conda remove --name tf --all #删除tf环境(全部删除)
将会自动安装如下组件
numpy
wheel
tensorflow-tensorboard
six
protobuf
html5lib
markdown
werkzeug
bleach
setuptools
使用下列代码测试安装正确性
6.测试.
命令行下:
activate tf
python
输入以下代码:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
没有报错就是配置好了。