文档来源转载:
http://blog.csdn.net/u010099080/article/details/53418159
http://blog.nitishmutha.com/tensorflow/2017/01/22/TensorFlow-with-gpu-for-windows.html
安装前准备
TensorFlow 有两个版本:CPU 版本和 GPU 版本。GPU 版本需要 CUDA 和 cuDNN 的支持,CPU 版本不需要。如果你要安装 GPU 版本,请先确认你的显卡支持 CUDA。我安装的是 GPU 版本,采用 pip 安装方式,所以就以 GPU 安装为例,CPU 版本只不过不需要安装 CUDA 和 cuDNN。
- 在 这里 确认你的显卡支持 CUDA。
- 确保你的 Python 版本是 3.5 64 位及以上。(TensorFlow 从 1.2 开始支持 Python 3.6,之前的官方是不支持的)
- 确保你有稳定的网络连接。
- 确保你的 pip 版本 >= 8.1。用
pip -V
查看当前pip
版本,用python -m pip install -U pip
升级pip
。 确保你安装了 VS2015 或者 2013 或者 2010。此条非必须,删除。
此外,建议安装 Anaconda,因为这个集成了很多科学计算所必需的库,能够避免很多依赖问题,安装教程可以参考 这里。
以上条件符合,那么恭喜你可以开始下载 CUDA 和 cuDNN 的安装包了,注意版本号分别是 CUDA 8.0 和 cuDNN 5.1(由于 TensorFlow 不同版本有变化,5.1 已不再适用新版本,此处请结合 安装cuDNN 说明),这是 Google 官方推荐的。可以去各自官网下载,我已经下载好打成一个压缩包放到了百度云,大家可以从 这里下载,密码 5aoc。
安装TensorFlow
由于Google那帮人已经把 TensorFlow 打成了一个 pip 安装包,所以现在可以用正常安装包的方式安装 TensorFlow 了,就是进入命令行执行下面这一条简单的语句:
# GPU版本
pip3 install --upgrade tensorflow-gpu
# CPU版本
pip3 install --upgrade tensorflow
- 1
- 2
- 3
- 4
- 5
然后就开始安装了,速度视网速而定。
安装网之后你试着在 Python 中import tensorflow
会告诉你没有找到 CUDA 和 cuDNN,所以下一步就是安装这两个东西。
安装CUDA 8.0
下载并安装CUDA Toolkit
Toolkit 8.0或更高版本:https://developer.nvidia.com/cuda-downloads
示例安装目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
这个也是很简单的,下载完我上面给的压缩包之后,解压,得到两个文件,那个 exe 文件就是 CUDA8 的安装程序,直接双击执行就可以了,就像安装正常的其他软件一样,安装过程屏幕可能会闪烁,不要紧,而且安装时间有点长。
安装完之后系统变量会自动为你添加上,这个不用管。
测试一下是否安装成功,命令行输入 nvcc -V
,看到版本信息就表示安装成功了。
安装cuDNN
Google 于 2017 年 8 月 17 日左右发布了 TensorFlow 1.3,该版本不再支持 cuDNN 5,开始支持 cuDNN 6,并预计在 TensorFlow 1.4 支持 cuDNN 7,所以说当你使用 pip 安装最新版的时候,请使用 cuDNN 6,而不是我提供的 5.1,否则会出现 Issues #2 的问题。
TensorFlow 1.3 中关于 cuDNN 的 发行说明:
All our prebuilt binaries have been built with cuDNN 6. We anticipate releasing TensorFlow 1.4 with cuDNN 7.
TensorFlow 1.2 中关于 cuDNN 的 发行说明:
TensorFlow 1.2 may be the last time we build with cuDNN 5.1. Starting with TensorFlow 1.3, we will try to build all our prebuilt binaries with cuDNN 6.0. While we will try to keep our source code compatible with cuDNN 5.1, it will be best effort.
下载并安装适用
于Windows 10的cuDNN cuDNN 6.1版库:https://developer.nvidia.com/cudnn
您需要在Nvidia注册才能下载这些文件。现在将cuDNN文件解压到您的Toolkit目录中
环境变量
确保在安装CUDA工具包之后,将CUDA_HOME设置为环境变量。如果没有,那么你需要手动添加它。
和路径变量一样..
安装Anaconda
我们将安装Anaconda,因为它可以帮助我们轻松管理特定Python发行版的单独环境,而不会影响系统中安装的python版本。
下载并安装
Anaconda
创建conda环境
创建新的环境,名称为tensorflow-gpu和python版本3.5.2conda create -n tensorflow-gpu python=3.5.2
激活环境 activate tensorflow-gpu
安装tensorFlowpip install
https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow_gpu-1.3.0rc0-cp35-cp35m-win_amd64.whl
这里我是在清华源网安装的tensorflow-gpu 的1.3版本,定向版本安装。
完成。您已经在您的Windows机器上成功安装了带有GPU的TensorFlow。
现在让我们来测试它是否使用GPU ...
激活我们创建的环境activate tensorflow-gpu
测试GPU
进入python shellpython
import tensorflow as tf
现在运行这个命令并检查它是否识别你的GPU。sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
Issues
#1 Cannot remove entries from nonexistent file
如果在安装 TensorFlow 的时候出现类似 Cannot remove entries from nonexistent file c:\users\li\anaconda3\lib\site-packages\easy-install.pth
的错误,那么可以参考 Cannot remove entries from nonexistent #622 和 osx 10.11 installation issues #135,里面说了好多种解决办法,我在这里介绍一种方法:在 pip3 install --upgrade tensorflow-gpu
之前先执行 pip install --upgrade --ignore-installed setuptools
。
#2 ImportError: DLL load failed: 找不到指定的模块。
和 ImportError: No module named ‘_pywrap_tensorflow_internal‘
如果在 import tensorflow
的时候这两个问题同时出现,那么很有可能是你的 cuda 和 cudnn 版本有问题,例如你的 cuda 版本是 8.0.60
,而正确的是 8.0.44
,重新安装正确的版本(文章里提供的)就可以。参考 On Windows, running “import tensorflow” generates No module named “_pywrap_tensorflow” error 。感谢 @qq_27690673 提供的信息。
#3 ImportError: No module named ‘tensorflow.python.pywrap_tensorflow_internal‘
如果在 import tensorflow
的时候出现此问题,那么你可能是在 tensorflow 的源码目录里进入了 python 解释器。离开该目录重新进入 python 解释器即可。
原文地址:https://www.cnblogs.com/Davirain/p/8452089.html