上海涛德顾问学院

TensorFlow培训系列文章:GPU版的TensorFlow深度学习平台安装

摘要: GPU版的TensorFlow如何在Windows安装作者:涛德原创Python 安装 1 安装Python 3.5 采用 Anaconda 3-4.2.0 本教材编写时,TensorFlow 1暂时在WINDOWS上不支持Python 3.5以后版本 https://repo.continuum.io/archive/ ...

GPU版的TensorFlow如何在Windows安装

 

作者:涛德原创

 

Python 安装

 

安装Python 3.5 采用 Anaconda 3-4.2.0

本教材编写时,TensorFlow 1暂时在WINDOWS上不支持Python 3.5以后版本。(注:目前已经支持后续版本)

https://repo.continuum.io/archive/Anaconda3-4.2.0-Windows-x86_64.exe

 

执行安装程序,选择安装目录,例如

D:\Anaconda3

 

安装TensorFlow

 

首先进入 Anaconda Prompt

 

1.       安装清华大学的conda

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

 

2.       再次执行,确保这个源在源列表最前面

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

 

3.       设置显示下载包的源

conda config --set show_channel_urls yes

 

4.        激活默认环境

activate root

 

5.       安装用 conda最新pandas

conda install pandas

 

6.       安装tensorflow

本教材编写时 conda暂时没有tensorflow的包(注:当前已支持)

不能用:conda install  tensorflow 安装,我们采用传统的pip安装

注:--upgrade --ignore-installed 是因为一个安装BUG,在当前版本中必须加上。

 

pip install --upgrade --ignore-installed tensorflow

 

另外:安装GPU版本执行下面命令

pip install --upgrade --ignore-installed tensorflow-gpu

 

配置GPU:

 

CUDA安装


我们虽然已经安装好了TensorFlow,但是如果需要GPU加速,我们还需要安装cudacuDnn(专门为deep learning准备的加速库)

 

1. cuda v8.x安装包,根据需要选择操作系统

下载路径:https://developer.nvidia.com/cuda-downloads

 

2. 下载cuDnn

https://developer.nvidia.com/cudnn

注册并且填问卷以后,直接拷到cuda路径对应的文件夹下面

3.cuda 环境变量设置

cuda安装好后

默认的环境变量配置不全,默认CUDA_PATH\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0,但是这样不能直接访问到binlib\x64下的程序包,在需要在path中加上binlib\x64的路径

 

4.TensorFlow测试

 

import tensorflow as tf

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

sess = tf.Session()

 



相关阅读

联系

上海涛德顾问学院 ( 沪ICP备14006824号 )  

GMT+8, 2018-12-11 11:41 , Processed in 0.140668 second(s), 15 queries , Gzip On.

Top Data World

回顶部