win10下基于anaconda安装GPU版本的tensorflow1.8.0 cuda9.0 cudnn9.0

cudnn9安装包: https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda_9.0.176_win10-exe

环境:win1064位)

GPU: GTX1080Ti

CUDA:9.0

Cudnn:9.0  (https://download.csdn.net/download/weixin_34613450/10641115

Tensorflow-gpu:1.8.0

 

第一步:安装Anaconda

Anaconda里面集成了很多关于python科学计算的第三方库,主要是安装方便,在官网(https://www.continuum.io/downloads)上即可下载。

win10下基于anaconda安装GPU版本的tensorflow1.8.0 cuda9.0 cudnn9.0_第1张图片

由于在WindowsTensorFlow目前只支持Python3.5 64-bit版本,为了安装方便,建议安装anaconda4.2.0版本,若为其他版本,需conda创建一个Python3.5的环境。

目前anaconda已更新至4.4.0版本,若需安装anaconda 4.2.0版本,在图示页面点击:Use thesezippedWindows installers,弹出了anaconda的历史版本供下载,选择相应版本即可。

win10下基于anaconda安装GPU版本的tensorflow1.8.0 cuda9.0 cudnn9.0_第2张图片

win10下基于anaconda安装GPU版本的tensorflow1.8.0 cuda9.0 cudnn9.0_第3张图片

第二步:安装CUDA

win10下基于anaconda安装GPU版本的tensorflow1.8.0 cuda9.0 cudnn9.0_第4张图片

 

CUDAnetwork版本与local版本,建议下载local版本,因为network版本是在安装过程中下载的,选择本地版较好,不过在官网下载容易出错(我下了几次均是下载到一半提示任务出错),建议在百度云下载。

下载之后安装即可。

win10下基于anaconda安装GPU版本的tensorflow1.8.0 cuda9.0 cudnn9.0_第5张图片

第三步:下载cudnn

初次下载cudnn需要在网上填写一堆问卷,耐心填吧。(下载地址:https://developer.nvidia.com/cudnn)下载好后无须安装。也可以在我的资源上下载:https://download.csdn.net/download/weixin_34613450/10641115

将包内三个文件复制到cuda相应位置,如图示。

win10下基于anaconda安装GPU版本的tensorflow1.8.0 cuda9.0 cudnn9.0_第6张图片

 win10下基于anaconda安装GPU版本的tensorflow1.8.0 cuda9.0 cudnn9.0_第7张图片

同时还需将bin目录、libnvvp目录添加到环境变量path中,否则后面调用时会报错。

win10下基于anaconda安装GPU版本的tensorflow1.8.0 cuda9.0 cudnn9.0_第8张图片

第四步:安装TensorflowGPU

        打开Anaconda Prompt,输入:pip install tensorflow-gpu=1.8.0安装即可。如果网速太慢,可以加载国内镜像,使用如下:

pip install tensorflow-gpu=1.8.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

win10下基于anaconda安装GPU版本的tensorflow1.8.0 cuda9.0 cudnn9.0_第9张图片

如果提示pip版本需更新,按提示操作即可。

 

经过上面几步,TensorFlow的GPU版本就安装好了,打开Anaconda Prompt,输入python或spyder或jupyter notebook即可开车了~~~。

下面验证下是否成功:

import tensorflow as tf

 

hello =tf.constant("Hello!TensorFlow")

sess = tf.Session()

print(sess.run(hello))

 

#Creates a graph.

a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0,6.0], shape=[2, 3], name='a')

b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0,6.0], shape=[3, 2], name='b')

c = tf.matmul(a, b)

 

#Creates a session withlog_device_placement set to True.

sess =tf.Session(config=tf.ConfigProto(log_device_placement=True))

#Runs the op.

print(sess.run(c))

win10下基于anaconda安装GPU版本的tensorflow1.8.0 cuda9.0 cudnn9.0_第10张图片

没有报错,安装成功~~~

你可能感兴趣的:(深度学习,tensorflow)