Welcome(〃'▽'〃)!

这里可能没有什么厉害的技术帖,但是会有一些实用的小教程

Anaconda+CUDA+Tensorflow(GPU)安装教程(Windows10下)

昨天博主的电脑在windows自动更新后出现了难以解决的bug(哭),没办法只好重装系统,重装系统一时爽,之后恢复各种软件环境才是最难受的,尤其是GPU版本的Tensorflow,N个月以前装的时候没想到写一个教程,结果这回装起来又是踩了各种坑,所以这次还是要写个教程,这样以后重新配环境可以直接看自己的博客(不过希望还是不要再重装了吧)

《Anaconda+CUDA+Tensorflow(GPU)安装教程(Windows10下)》

本教程的环境为Windows10+Anaconda,首先安装Anaconda,由于tensorflow最新就支持到python3.6,所以我们不能直接从官网上下载最新的Anaconda,这里推荐一下清华的一个镜像网站,可以下载Anaconda曾经的版本,各版本Anaconda和python版本的对应关系如下:

《Anaconda+CUDA+Tensorflow(GPU)安装教程(Windows10下)》

我们就选择Anaconda3-5.2.0-Windows-x86_64.exe,这样python版本为3.6,可以支持tensorflow。

下载后打开安装包,除了一个选择安装位置的地方,不停地点下一步就行。

Anaconda安装完毕,接下来是重头戏,安装CUDA和cudnn,tensorflow现在还不支持CUDA10,所以我们选择CUDA9.0,配套的是cudnn7(提醒:安装前请关闭杀毒软件)

CUDA9.0下载地址:https://developer.nvidia.com/cuda-90-download-archive

《Anaconda+CUDA+Tensorflow(GPU)安装教程(Windows10下)》

cudnn7下载地址:https://developer.nvidia.com/cudnn,这里是需要注册填问卷才能下载,比较麻烦,可以直接从我的百度网盘下载:

链接:https://pan.baidu.com/s/1NFlQ871uoAMmu5uv58Zs9g
提取码:3j99

接下来打开CUDA9.0的安装程序,首先会要求你选择一个文件夹把安装文件暂时存放在里面,接着开始安装,建议选择自定义安装,把Visual Studio Integration的勾选去掉,否则大概率会报错(原因是和最新版本的vs2017不兼容,这个我在以前安装的时候没遇到,可能是因为以前装的vs2017版本比较久),然后如果你的驱动比安装包里面的新的话,driver components的选项也可以去掉(下图是网络上找的图)

《Anaconda+CUDA+Tensorflow(GPU)安装教程(Windows10下)》

接着就直接下一步开始安装,一般不会有问题了

安装完成后因为我们之前没装 Visual Studio Integration ,所以还要做以下操作:

把安装程序解压到一个文件夹中,找到CUDAVisualStudioIntegration\extras\visual_studio_integration\MSBuildExtensions ,复制里面的所有文件,然后打开自己的 VS2017 的安装目录,比如 C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\Common7\IDE\VC\VCTargets\BuildCustomizations,把文件复制进来即可

接下来安装cudnn7,解压缩以后将里面的三个文件夹bin,include,lib复制到cuda安装目录中,如下图

《Anaconda+CUDA+Tensorflow(GPU)安装教程(Windows10下)》
cudnn7内的文件
《Anaconda+CUDA+Tensorflow(GPU)安装教程(Windows10下)》
复制到的位置

到这步为止,我们完成了最麻烦的工作,接下来就是安装tensorflow-gpu:

打开Anaconda Prompt的窗口,输入

pip install tensorflow-gpu==1.9.0

等待安装完毕即可(注意:必须加上1.9.0这个版本号,因为之后的tensorflow版本和cudnn7可能存在兼容问题,这里我当初踩了坑,还有就是建议挂代理,否则比较慢)

大功告成!我们写一段简单的程序验证一下是否安装成功:

import tensorflow as tf
hello = tf.constant('Hello, tensorflow!')
sess = tf.Session()
print(sess.run(hello))
a = tf.constant(1)
b = tf.constant(2)
c =sess.run(a+b)
print(c)

输出:

b'Hello, tensorflow!'
3

安装完毕撒花!对了,既然tensorflow都装了,就顺便把pytorch也装一下吧,直接在Anaconda Prompt中输入:

conda install pytorch torchvision -c pytorch

询问是否安装,输入y回车即可

可能遇到的问题1:安装tensorflow过程中可能会遇到msgpack未安装的warning,在命令行中输入:

pip install msgpack

即可

可能遇到的问题2:运行时如果有FutureWarning: Conversion of the second argument of issubdtype from float to np.floating is deprecated的warning的话,可能是h5py版本不对,我们在anaconda prompt中输入 :

pip install h5py==2.8.0rc1 

即可

点赞

发表评论

电子邮件地址不会被公开。