在本文的tensorflow的环境搭建中,所需要提前说明的是,我的安装配置的tensorflow-gpu的版本是1.10.0,且相应的cuda的版本是,提前说明的原因是tensorflow-gpu的安装版本和cuda以及cudnn的版本必须对应且正确。以下则是tensorflow-gpu不同版本的对应的cuda9.0和对应的cudnn7.3的说明图,在这里只提到了GPU版本的配置,相关的CPU版本的配置就不做说明了,网上也有许多,配置过程也相对简单一些:
1.所需安装程序和文件
- CUDA 9.0-windows10-x64版本
- cudnn cudnn-9.0-windows10-x64-7.3.0版本
- Anaconda3-5.0.1-Windows-x86_64
2.下载相关的包
2.1 下载Anaconda
下载Anaconda3-5.0.1,在Anaconda官网下载或者用其他方式下载Anaconda3-5.0.1,这个下载的anaconda对应的python版本是3.5,现在一般配置tensorflow-gpu对应的python版本是3.5以上,且一般是python3.5和python3.6版本。
安装anaconda的时候,需要勾选第一项,如下图所示
其他的安装步骤则是简单的下一步,下一步即可完成。
安装完成的时候,由于勾选的第一项,会在环境变量中有如下图的显示,证明anaconda的环境变量是自己配置好的,如下:
2.2 下载CUDA
咱们来下载CUDA啊,移步NVIDIA下载网站:https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal,下载选项看下图:
2.3 下载cudnn
下载cuda后,在https://developer.nvidia.com/rdp/form/cudnn-download-survey下准备下载cudnn对应的版本,下载时需要注册登录一个账号,,登陆进去后选择对应版本,如下图所示
3.相关安装包和文件的配置
3.1 CUDA的安装过程
x
在进行到选项这里的时候,一般要勾选掉Visual Studio,一般才会顺利进行下一步的安装。
安装效果如下所示则为成功,然后重新启动计算机则进行下一步
3.2 cudnn的下载
下载完后解压得到三个文件夹如下图所示:
将三个文件夹下的对应的文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0下的对应的bin,include,lib目录
然后配置环境变量,安装成功后有:
然后配置path,如下图所示,有两个路径了,还需要添加
这样环境变量就配置成功了。
4. 在cmd下安装配置相应的包:
4.1 安装tensorflow-gpu
pip install -i https://pypi.douban.com/simple/ tensorflow-gpu==1.10
执行此命令,在cmd之中,会下载tensorflow-gpu很快,且不会因为下载速度太慢而失败。且记tensorflow-gpu前面是有一个空格的。且tensorflow-gpu必须为1.10版本。记:等号必须是两个== 。若安装失败,可以在下面网址直接下载对应的包下载后pip install 拖到后面去就行,若有不懂的,可以联系我:QQ:347197038。
https://www.lfd.uci.edu/~gohlke/pythonlibs/
下载对应的版本,记得:例如,我的电脑是64位的,且我的anaconda安装的python是3.6版本,,所以对应的是下面这个依赖库,下载后,直接pip install 后将依赖文件库拖入到cmd中即可
4.2 安装opencv-python
这个是本人学习图像处理方面的东西需要的包,做深度学习方面的东西也是和图像处理相关的,所以需要用到opencv。这个不是必须要安装的
注:这个很重要,是后面做实验发现的,就是opencv-python的版本问题,opencv库在不停更新,更新了很多需要的东西,但是导致很多东西的用法不一样了,比如说resize这个函数,现在都更新到opencv4的版本了,所以我在做实验的时候就出问题了,跟以前的代码写法有所不同,制定opencv-python的版本的时候注意如下,这个版本必须是eg:3.4.0.14,后缀要复杂一点,要注意:
4.3 安装keras
pip install -i https://pypi.douban.com/simple/ keras
同样的道理,运用豆瓣源安装是很快的,且记住有空格,这里keras的版本就不是很重要了,不是需要注意的事项。
5检测tensorflow-gpu是否安装成功
用图说明,这个比较简单,就不用文字说明了,看一下下图吧。
谢谢大家指教,这个是完整的版本安装和配置,有什么建议,可以留言。谢谢!
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:windows下简单搭建Tensorflow,keras(GPU)环境 - Python技术站