配置机器学习训练环境太麻烦?开源工具Parris说一键就搞定

本文来自AI新媒体量子位(QbitAI)

在搭建、训练机器学习模型的过程中,你的时间可能大部分都花在了那些与算法无关的事情上:收集数据、清洗数据、标注,甚至基础环境的配置,也需要不少时间。

配置机器学习训练环境太麻烦?开源工具Parris说一键就搞定

有一个名叫Parris的开源工具,就要让配置基础环境这个环节自动化。今天,它刚被发布到Reddit论坛上就收获了不少赞。

据作者介绍,这个工具针对云端的机器学习训练,一键完成配置服务器、安装操作系统、安装GPU驱动、为项目安装各种依赖包等等环节,同时加载要训练的模型和训练集。

另外,用户还可以为服务器设置一个自动关机时间。

配置机器学习训练环境太麻烦?开源工具Parris说一键就搞定

Parris目前只支持亚马逊AWS,依靠CloudFormation模板来实现一键启动训练,需要用户有AWS账户和证书。不过,据作者说,以后的版本打算支持Google云、微软Azure等等。

要使用Parris,你需要先准备好要训练的模型和数据集,安装方式如下:

配置机器学习训练环境太麻烦?开源工具Parris说一键就搞定

GitHub地址:https://github.com/jgreenemi/Parris

本文作者:夏乙
原文发布时间:2018-01-02