Windows10+TensorFlow 1.8 GPU版本安装

第一步,在Tensorflow官网,阅读安装指南。https://www.tensorflow.org/install/install_windows

在官网阅读安装指南,是最省心省力的方法,比用baidu找各种不靠谱的文章节约时间多了。

安装指南中明确说明,需要:

1,CUDA Toolkit 9.0, 注意,版本一定要是V9.0,不能9.1,9.2...

2,对应CUDA toolkit 9.0的cuDNN v7.4.2

每次安装前,希望大家都先去Tensorflow官网阅读安装指南,然后再动手。


Windows10+TensorFlow 1.8 GPU版本安装_第1张图片

第二步,去nvidia官网下载CUDA 9.0 和对应的cuDNN

CUDA Toolkit 9.0下载地址:https://developer.nvidia.com/cuda-90-download-archive;注意,根据自己的操作系统,选择相应的版本,按照默认选项安装即可。

Windows10+TensorFlow 1.8 GPU版本安装_第2张图片
CUDA 

cuDNN 下载地址:https://developer.nvidia.com/rdp/cudnn-download;注意选择与CUDA Toolkit V9.0对应的版本。

Windows10+TensorFlow 1.8 GPU版本安装_第3张图片

下载解压后,把cudnn64_7.dll,cudnn.h,cudnn.lib 拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 对应文件下

Windows10+TensorFlow 1.8 GPU版本安装_第4张图片

这样就完成了GPU相关库的安装。

第三步,下载并安装Anaconda:https://www.anaconda.com/download/。安装完毕后,打开Anaconda navigator, 在Terminal里面,键入命令:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --ignore-installed --upgrade tensorflow-gpu,自动安装tensorflow gpu 版以及相关的依赖库,如下图所示

Windows10+TensorFlow 1.8 GPU版本安装_第5张图片
开打Terminal
Windows10+TensorFlow 1.8 GPU版本安装_第6张图片
安装 tensorflow GPU 版

最后一步,在Terminal里面键入如下命令,测试tensorflow gpu版是否安装成功。

python

>>> import tensorflow as tf

>>> hello = tf.constant('Hello, TensorFlow!')

>>> sess = tf.Session()

>>> print(sess.run(hello))

若跟下图类似,即安装成功


Windows10+TensorFlow 1.8 GPU版本安装_第7张图片

安装出错:若版本对应,按照上述步骤,基本上是一次性安装成功。另外,官网的安装指南里面,有常见安装错误说明,如下图

Windows10+TensorFlow 1.8 GPU版本安装_第8张图片
Tensorflow常见安装问题

参考阅读:《深度学习图像识别技术》

你可能感兴趣的:(Windows10+TensorFlow 1.8 GPU版本安装)