TensorFlow基本上成为了深度学习的标配。但是为了学习主要的内容(忽略底层实现细节)或者快速实现一个神经网络模型,Keras是一个很好的选择。
环境准备
安装步骤
安装R
的Keras
需要用到devtools::install_github()
命令,因此需要安装devtools
包。
但是笔者用的Windows 10
,直接安装包不会报错,但是使用该包下的命令会出一些奇怪的错误,例如在装R-Keras
时就报超时。
所以建议在Windows
下使用该包时,先去安装Rtools。
同时需要安装好Anaconda
,不然在后续步骤中会报错。
安装好Rtools
之后,打开R命令行
,在其中再安装一遍devtools
包:
1 | install.packages("devtools") |
然后安装R
的Keras
接口:
1 | devtools::install_github("rstudio/keras") |
在命令行或者第一次运行Keras
的脚本下输入:
1 | library(keras) |
然后会调用Anaconda
为R-Keras
建立一个环境,成功初始化,在以后的R脚本
就不需要再使用install_keras()
命令了。
安装GPU支持
事实上这部分操作是属于Anaconda
那边的操作,毕竟R-Keras
用的还是Python
的TensorFlow
后端。
中文文档里面就已经很详细地介绍了如何准备GPU
环境:
文档写的时候上述Keras尚未支持高版本的CUDA
和cuDNN
,经笔者测试,是可以支持最新版的CUDA
和cuDNN
的。
安装好后,打开Anaconda Navigator
管理软件,或者命令行模式:
1 | $ activate r-tensorflow |
切换到r-tensorflow
环境,安装gpu
版Keras
:
1 | $ conda install keras-gpu |
等待安装过程(可能因为墙导致下载出问题,要么上梯子,要么辛苦一点重试几次安装命令吧)后,就可以开始使用了。
测试R-Keras
官网给了MNIST
测试集作为Demo,然而AWS
被墙了,整个Demo直接卡在下载数据集步骤就挂了。
笔者写了个简单的脚本用于测试:
1 | library(keras) |
如果脚本成功运行的话,会在相同目录下产生一份名为history.png
的图片,那么恭喜,安装成功了。