欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Windows Tensorflow 环境搭建

程序员文章站 2022-03-29 19:40:54
...

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 环境搭建