踩坑tensorflow的gpu安装(win10)

1.忙活了一上午,终于搞定了。在安装前一定要安装好vs,我安装的是vs2012,我对vs2012有一种特殊的执着。
然后是首先安装anaconda,教程都有,大家基本都不会掉在这个坑里,我就跳过了。
2.安装CUDA和CuDNN,我是安装了CUDA9.0和CuDNN7.0的,这只是手动安装的一部分,其实后来安装tensorflow的时候也会安装一遍,不过为了保险,先装吧。
3.安装CUDA遇到问题是因为之前装了别的版本的,把别的版本的删了就行,一般在C盘,把C:\Program Files里的NVIDIA Corporation和NVIDIA GPU Computing Toolkit删了就行,再安装就没问题了。
4.解压cudnn的压缩包里面有三个文件夹
在这里插入图片描述
把这三个文件夹复制到你cuda的安装目录下,我的地址是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
在这里插入图片描述

这样CUDA和CuDNN就安装好了。
5.创建tensorflow环境,打开anaconda prompt
通过调用以下命令创建名为 tensorflow 的 conda 环境:
C:> conda create -n tensorflow pip python=3.6(我的python是3.6的)
通过发出以下命令激活 conda 环境:
C:> activate tensorflow
现在,基本环境已经配置好了,我们要安装一些重要的Python科学运算库,Anaconda已经为我们准备好的一系列常用的Python苦,例如numpy,pandas,matplotlib等等,所以我们只需要安装一次anaconda库就可以把这些库全部安装好。
conda install anaconda
6.安装tensorflow
conda install tensorflow-gpu
就会自动安装gpu版本的最新tensorflow
7.测试,在anaconda prompt里输入python再回车进入python交互式环境,依次输入每一句然后回车
import tensorflow as tf
hello = tf.constant(‘Hello,TensorFlow!’)
sess = tf.Session()
print(sess.run(hello))
最后如果显示了GPU信息,就说明安装成功啦!

猜你喜欢

转载自blog.csdn.net/zhikaixuan/article/details/83309156