Windows 10 - 64位,GeForce GTX 1070, Visual Studio 2015,Anaconda 3, Cuda10.0,cuDNN 7.4.2, Tensorflow1.13.1-GPU版本
配置了两天把Tensorflow需要的编译环境配置成功并运行了tensorflow的样例程序,还是有些艰辛的。
参考文献: https://www.cnblogs.com/guoyaohua/p/9265268.html
根据自己的电脑进行了一些调整,期间也踩了一些坑,万幸都解决了,现在在这里总结整理并分享一下,这样大家在遇到相同问题时就可以轻松通过啦。^ ^
Visual Studio 2015下载地址:https://www.microsoft.com/zh-cn/download/details.aspx?id=48145
如果官网下载不了,可以在我的百度云盘下载,下载地址:https://pan.baidu.com/s/19bjb6YneRc0u-jYUuaNfiw 密码:bb35
一定要在先安装Visual Studio 2015,这样在安装Cuda的时候就会关联VS2015,省去很多的麻烦。
官网地址:https://www.anaconda.com/download/
进入上边的链接,选择Windows,选择Windows,然后根据自己的系统位数选择Python 3.7 version的installer。这里下载的Python版本是3.7不过在后面可以更改python的版本。
我们之后需要手动添加环境变量,所以“Add Anaconda to my PATH environment variable”这里不要勾选。
安装完成之后,打开系统的环境变量,流程是 右键这台电脑→属性→高级系统设置→环境变量,会进入到下面的这个页面。
双击用户变量的Path,添加以下路径。(注意这里要换成你自己的安装路径)
C:\Users\Fuhao\Anaconda3
C:\Users\Fuhao\Anaconda3\Scripts
C:\Users\Fuhao\Anaconda3\Library\bin
结束后我们来打开cmd测试一下,在cmd中打入 conda -V,得到的是当前anaconda的版本号,证明配置成功。
cuda10.0 下载地址:https://developer.nvidia.com/cuda-10.0-download-archive
cuDNN7.4.2 下载地址:https://developer.nvidia.com/rdp/cudnn-archive
用之前的方法打开环境变量,双击用户变量的Path,添加以下路径。(注意这里要换成你自己的安装路径)
首先我们要创建一个编译环境,调用下列命令,来创建一个名称为tensorflow_gpu的环境:
conda create -n tensorflow_gpu pip python=3.5
这里输入y,说明允许安装,之后就是等待,当出现下列提示时说明已经创建完毕。
通过之前的提示说明,激活环境是在命令行中输入 activate tensorflow
,而关闭环境则为deactivate
.由于我们的环境名称为 tensorflow_gpu,输入下列命令来激活我们创建好的conda环境。
activate tensorflow_gpu
我们现在已经进入了我们创建好的“tensorflow_gpu”的环境。
这里分别给出安装GPU和CPU版本tensorflow的命令。
GPU版本:
pip install --ignore-installed --upgrade tensorflow-gpu
CPU版本:
pip install --ignore-installed --upgrade tensorflow
出现以上提示就代表我们的tensorflow-GPU版本已经安装完毕了。不过我们是在我们创建好的环境中安装了tensorflow,并不是在base中,而anaconda是安装在base中的,所以比如我们打开jupyter notebook来写代码,会发现我们使用的kernel是base中的kernel,但是base中没有tensorflow,所以import会报错。所以我们需要在我们的环境里也安装anaconda的基础包。用下面的命令:
conda install anaconda
在我们的环境中anaconda安装完毕后,理论上来说就完成了所有的安装了,但是在这里我出现了一个小bug,就是在import tensorflow as tf 的时候cmd会闪退,没有出现这种情况的同学那么恭喜你你已经完成了tensorflow-GPU版本的环境搭建和配置,直接忽略这一段往下看测试环节,那与我出现同样问题的同学下面是解决方法。进入你创建的环境中的site-packages,如C:\Users\Fuhao\Anaconda\envs\tensorflow_gpu(你创建的环境名字)\Lib\site-packages,删除 h5py 和 h5py-2.7.1-py3.5.egg-info 这两个文件夹(注意:还有一个h5py-2.9.0.dist-info不要删,只删除之前的那两个),重新打开cmd,再次尝试import的时候就不会再闪退了。
最后我们来进行简单的测试,进入cmd,激活我们的环境,然后键入Python进入python解释器模式,调用下列命令:
import tensorflow as tf
没有报错说明我们import成功,接下来调用下列命令来检查是否使用GPU:
tf.test.gpu_device_name()
出现下列提示说明GPU已被使用。
至此,对Tensorflow的环境搭建,安装和测试就全部结束了,希望大家都可以成功的安装。^ ^
如果有什么问题,可以和我一块交流和学习~