Windows Tensorflow 环境搭建

2017-12-20 13:58:38来源:CSDN作者:ja33son人点击

分享

Windows Tensorflow 环境搭建

python 安装

推荐安装3.6.3版本,下载以后,一直next即可。
下载地址:
https://www.python.org/ftp/python/3.6.3/python-3.6.3-amd64.exe

安装后的测试方法,cmd,python,显示Python版本号为3.6.3

python 验证

TensorFlow 安装

使用国内源能有效改善下载速度,此处使用的是豆瓣的源
打开cmd,输入以下命令即可。

pip install -i https://pypi.douban.com/simple/ tensorflow

如果硬件有N卡(NVIDIA显卡)支持,想要安装GPU版本的TensorFlow,命令为

pip install -i https://pypi.douban.com/simple/ tensorflow-gpu

并且要安装N卡驱动以及cuda和cuDNN
下载地址:
cuda:https://developer.nvidia.com/cuda-toolkit-archive
cuDNN:https://developer.nvidia.com/cudnn
注意:
cuDNN的下载需要先注册账号,此处不再赘述。
TensorFlow目前支持cuda的版本为cuda8,不要安装最新的cuda9。
cuDNN的下载也要对应cuda版本。

cuDNN下载完成后,将lib复制到cuda的安装目录下即可。

以上步骤完成后,TensorFlow环境就搭建完成了,很简单吧。
测试一下:
cmd

Pythonimport tensorflow as tf

没有报错,证明TensorFlow CPU环境搭建成功

Pythonimport tensorflow-gpu as tf

没有报错,证明TensorFlow GPU环境搭建成功

测试效果图

最新文章

123

最新摄影

闪念基因

微信扫一扫

第七城市微信公众平台