Keras相对于tensorflow用起来要爽很多。使用GPU版本的Keras,跑数据会比cpu版本的快很多。
前提条件:
环境:Windows10
显卡:Nvidia
如果是AMD的显卡,那就不用看这篇教程了。嘿嘿
话不多说,下面就是keras-GPU版本的配置。
1. 下载Anaconda。
使用andaconda的好处会给我们带来很多便利,比如,下载一个包,他会把这个包依赖的包都下载好。以及包含了一些好用的环境,比如jupyter notebook。
推荐从清华大学镜像站下载,会快很多。
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/?C=M&O=D
下载好对windows系统版本的安装包,这里不推荐下载最新的Anaconda,因为最新的Anaconda内置的是python 3.7,但是tensorflow现在支持的版本最高为3.6。如果你下载了,在后面有解决办法,但是不保证是通用。
然后直接下一步下一步下一步,安装好,安装的过程比较漫长,可以去干点别的。
等安装好了之后,配置一下环境变量。在path变量下,配置以下环境变量。