Tensorflow是当前最流行的机器学习框架之一,下面就为大家讲讲如何在windows上安装Tensorflow。
一、系统环境
TensorFlow是基于VC++2015开发的,所以需要下载安装VisualC++ Redistributable for Visual Studio 2015 来获取MSVCP140.DLL的支持。同时官方推荐的系统配置为:
- 64位 x86系统
- Windows7及以上版本
二、版本选择
TensorFlow支持CPU和GPU两种安装版本。CPU版本的环境需求简单,后者需要额外的支持。如果要安装GPU版本(电脑有NVIDIA显卡),需要以下额外环境:
- 下载安装CUDA Toolkit 9.0,并确保其路径添加到PATH环境变量里;
- CUDA9.0相关的NVIDIA驱动。
- 下载安装cuDNN v7.0,并确保其路径添加到PATH环境变量里;
- 有支持CUDA计算能力3.0或更高版本的NVIDIA GPU卡。
此外,如果电脑已经安装了上述多个版本的安装包,使用时一定要在环境变量中配置好或者删除不必要的版本。
三、安装方式
上一步完成之后,开始安装 TensorFlow。TensorFlow 可以通过两种方式进行安装,一是原生的 pip 命令安装,另一种是通过 Anaconda 安装。这里推荐第一种安装方式,比较简单并且简洁。
(一)使用原生 pip 命令安装
1)下载安装 Python 3.6.x 64-bit,要注意版本3.6.x。
2)打开 windows 的命令行窗口,安装CPU版本输入
pip install --upgrade tensorflow
安装GPU版本输入
pip install --upgrade tensorflow-gpu
注意:如果电脑同时安装有python2版本,上述pip需要改成pip3
(二)使用Anaconda安装
1)下载安装 Anaconda;
2)打开命令行窗口,创建conda环境
conda create -n tensorflow python=3.6
3)激活conda环境
activate tensorflow
4)安装CPU版本输入
pip install --ignore-installed --upgrade tensorflow
安装GPU版本输入
pip install --ignore-installed --upgrade tensorflow-gpu
三、验证安装
验证TensorFlow安装是否成功,可以在命令行窗口输入python进入python环境,或者运行相关IDE或编辑器之后输入以下代码:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
如果能正常输出hello字符串,则安装成功。
Hello, TensorFlow!
到此,大功告成,开始你的机器学习之旅吧!
注意:本文归作者所有,未经作者允许,不得转载