windows安装tensorflow快速教程

star2017 1年前 ⋅ 1075 阅读

  Tensorflow是当前最流行的机器学习框架之一,下面就为大家讲讲如何在windows上安装Tensorflow。


一、系统环境

  TensorFlow是基于VC++2015开发的,所以需要下载安装VisualC++ Redistributable for Visual Studio 2015 来获取MSVCP140.DLL的支持。同时官方推荐的系统配置为:

  1. 64位 x86系统
  2. Windows7及以上版本


二、版本选择

  TensorFlow支持CPU和GPU两种安装版本。CPU版本的环境需求简单,后者需要额外的支持。如果要安装GPU版本(电脑有NVIDIA显卡),需要以下额外环境:

  1. 下载安装CUDA Toolkit 9.0,并确保其路径添加到PATH环境变量里;
  2. CUDA9.0相关的NVIDIA驱动。
  3. 下载安装cuDNN v7.0,并确保其路径添加到PATH环境变量里;
  4. 有支持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!


  到此,大功告成,开始你的机器学习之旅吧!

更多内容请访问:IT源点

相关文章推荐

全部评论: 0

    我有话说: