首页 » Windows » Windows Tensorflow 环境搭建

Windows Tensorflow 环境搭建

原文 http://blog.csdn.net/ja33son/article/details/78851848

2017-12-21 02:10:04阅读(553)

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

Windows Tensorflow 环境搭建

Python环境变量

右键电脑 > 属性 > 高级系统设置 > 高级 > 环境变量
系统变量 > 新建
PY_HOME
你的Python路径
例如:
PY_HOME D:\test\Python363
添加后,在全局变量后添加以下环境变量,注意用;分割
%PY_HOME%\
%PY_HOME%\Scripts\

配置script环境变量,主要用来调用pip等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

Python
import tensorflow as tf

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

Python
import tensorflow-gpu as tf

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

Windows Tensorflow 环境搭建

最新发布

CentOS专题

关于本站

5ibc.net旗下博客站精品博文小部分原创、大部分从互联网收集整理。尊重作者版权、传播精品博文,让更多编程爱好者知晓!

小提示

按 Ctrl+D 键,
把本文加入收藏夹