Tensorflow的安装和测试

这些都是原来在微博上的文章,满满搬到这里吧

最近开始学习自然语言处理,用到一些神经网络模型需要使用tensorflow,于是在小本的win10系统上装了个CPU版本的tensorflow,按照这里的说明一步一步安装,没有遇到任何问题:https://stackoverflow.com/questions/34785414/how-to-install-tensorflow-on-windows/39902815#39902815

于是天真地以为tensorflow安装好简单啊,今天我要在实验室的工作站上装个GPU版本的,为了速度(bige)更高一些,结果折腾了整整一天,才终于搞定了。

所以还是把安装过程(caiguodekeng)记录在这里,备忘备忘

总结下来最最重要的一点就是版本!!!各个版本都要匹配!!!

一.python需要3.5版本

在Anaconda中安装并创建python3.5环境

conda create -n python35 python=3.5 anaconda

二.安装CUDA

我装的9.0版本,下载地址https://developer.nvidia.com/cuda-90-download-archive

三.安装cuDNN

下载地址https://developer.nvidia.com/rdp/cudnn-download

需要先注册登录,然后选择对应CUDA版本的cuDNN下载

压缩包解压后,将bin,include,lib三个文件合并复制到CUDA安装文件夹下:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0下

四.添加环境变量

我看了好几个安装教程,好多出错信息是因为环境变量没有添加完全,所以最后我把所有能添加的都添加到环境变量里了。。。

C:\Program Files\NVIDIA GPUComputing Toolkit\CUDA\v9.0

C:\Program Files\NVIDIA GPUComputing Toolkit\CUDA\v9.0\bin

C:\Program Files\NVIDIA GPU ComputingToolkit\CUDA\v9.0\libnvvp

C:\Program Files\NVIDIA GPU ComputingToolkit\CUDA\v9.0\include

C:\Program Files\NVIDIA GPU ComputingToolkit\CUDA\v9.0\lib

C:\Program Files\NVIDIA GPU ComputingToolkit\CUDA\v9.0\lib\x64

五.安装VS2015

下载地址https://www.microsoft.com/en-us/download/details.aspx?id=53587

六.Anaconda端口:conda activate python35

pip install tensorflow-gpu

七.安装完成,测试一下下面的代码

import tensorflowas tf

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

sess = tf.Session()

print(sess.run(hello))

然而我还是too young too simple,居然出现了Warning:

Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2

查了一下,其实也不影响程序运行,不过强迫症还是促使我找到了解决方案:

pip install h5py==2.8.0rc1

八.下面再测试一段

a = tf.constant(10)

b = tf.constant(32)

print(sess.run(a+b))

成功输出42(程序员为什么都喜欢用《银河系漫游指南》这个梗。。。)

你可能感兴趣的:(Tensorflow的安装和测试)