win10下autokeras的安装
按照官方网站的方法进行安装,注意:Auto-Keras只支持Python3.6。
项目github:https://github.com/jhfjhfj1/autokeras
pip install autokeras
autokeras需要的依赖项很多,包括tensorflow、pytorch、keras、numpy等,大部分在pip安装autokeras都可以自动下载,除了pytorch需要参考torch官网手动安装。总体而言,还是很简单的。
win10下autokeras的GPU使用
安装完成之后,就可以使用https://github.com/jhfjhfj1/autokeras/blob/master/examples/mnist.py文件进行测试:
from keras.datasets import mnist
from autokeras import Im