安装Anaconda
并配置安装Python3.5版本,因为TensorFlow在windows中只支持 Python3.5的版本
自己因为之前安装VS2017时候已经安装好了Anaconda,这里就不再安装,没安装Anaconda的,可以参考下面的博客:Win10 TensorFlow(gpu)安装详解
安装显卡驱动:
选择默认精简安装即可. Installs all CUDA components and overwrites current Display Driver.
安装CUDA:
下载:CUDA Toolkit 8.0 - Feb 2017
- 选择CUDA Toolkit 8.0 - Feb 2017,选择自己系统win10,选择本地安装。接着我们下载以下两个文件,
基础安装包 Base Installer 1.3GB
和补丁包 Patch 2 (Released Jun 26, 2017)
。下载好后,只需要双击运行安装第一个1.3GB的cuda_8.0.61_win10.exe ,都默认安装就行。
- 选择CUDA Toolkit 8.0 - Feb 2017,选择自己系统win10,选择本地安装。接着我们下载以下两个文件,
验证CUDA 8.0 是否 安装成功
- 打开cmd 输入:
nvcc -V
,若如下所示,则说明安装成功。
- 打开cmd 输入:
安装cuDNN:
将以上文件都下载好,解压后,分别将三个文件放到cuda的相应安装目录下:
C:\cuda\bin\cudnn64_5.dll —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
C:\cuda\include\cudnn.h —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include
C:\cuda\lib\x64\cudnn.lib —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64并将
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin\cudnn64_5.dll
配置到PATH环境变量中安装 TensorFlow
- 原生 pip 安装
如果没有安装Python3.5 ,需要将其先安装上。TensorFlow 在 Windows 上目前只支持Python 3.5.x。 注意 Python 3.5.x 使用 pip3,我们用 pip3 来安装 TensorFlow。
安装只支持 CPU 的 TensorFlow,在 terminal 中输入如下命令:
C:\> pip3 install --upgrade tensorflow
安装支持 GPU 的 TensorFlow,使用如下命令:
C:\> pip3 install --upgrade tensorflow-gpu
Anaconda 安装 ( 简单方便,自己推荐!)
(Anaconda 安装是社区支持,而非官方支持)
按说明下载并安装 Anaconda:
配置Anaconda环境变量于
PATH
中:
建立一个 命名为
tensorflowde
的conda 环境,指定Python版本是3.5 。通过打开系统命令行,Win+X+A
,然后再控制台输入:conda create -n tensorflow python=3.5
查看当前环境
conda info -e
可以看到,我们当前所在的是默认的root环境。
激活 anaconda 环境:
activate tensorflow (tensorflow)C:\> # 你的提示符应该发生变化
可以看到,我们已经激活tensorflow环境
然后在tensorflow 这个 conda 环境中安装只支持 CPU 的 TensorFlow(写在一行):
(tensorflow)C:\> pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.1.0-cp35-cp35m-win_amd64.whl
安装支持 GPU 的 TensorFlow(写在一行):
(tensorflow)C:\> pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.1.0-cp35-cp35m-win_amd64.whl
补充知识 conda基本命令 (具体细节可参考印象笔记)
- conda基本命令:
conda --version
查看当前Anaconda版本conda info -e
查看当前环境版本conda create -n python27 python=2.7
创建一个python2.7版本的开发环境activate python27
切换到 python2.7的版本分支(环境)deactivate
切换回 root分支(环境)-
conda remove -n python27 --all
删除一个分支(环境)conda create -n clonedpython27 --clone python27
克隆python27这个分支,并将新分支命名为clonedpython27
验证安装结果
- 安装完TensorFlow以后,我们继续在当前terminal中启动python。
python
- 在 Python 交互式环境中输入
>>> import tensorflow as tf >>> hello = tf.constant('Hello, TensorFlow!') >>> sess = tf.Session() >>> print(sess.run(hello))
- 如果系统输出如下,则安装成功:
Hello, TensorFlow!
在PyCharm中配置TensorFlow环境
要在pycharm下使用tensorflow,要设置好pycharm下解释器interpreter的路径
File --> Setting --> Project --> Project Interpreter
,这里也就是tensorflow的路径,如下图所示:
在PyCharm中测试该代码
>>> import tensorflow as tf >>> hello = tf.constant('Hello, TensorFlow!') >>> sess = tf.Session() >>> print(sess.run(hello))
运行后会提示警告,但这个警告并没有影响
W c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\core\platform\cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use FMA instructions, but these are available on your machine and could speed up CPU computations.
我们可以通过在上面代码之前,添加两条语句,忽略掉警告import os ##加上这两句,可以不显示警告 os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
Reference
Window10 安装 TensorFlow + CUDA8.0 + cudnn5.1
猜你喜欢
转载自blog.csdn.net/forever__1234/article/details/79845300
今日推荐
周排行