win10下autokeras的入门

本文介绍了在Windows 10系统下安装和使用Autokeras的步骤,包括按照官方方法安装,解决GPU使用问题,以及如何调整Autokeras的batch size。在安装时需要注意Autokeras仅支持Python 3.6,并且需要手动安装PyTorch。当发现GPU未被利用时,是由于utils.py中的get_device()函数问题,修改为适合Win10的命令即可使GPU正常工作。此外,若要改变batch size,可以直接修改constant.py文件中的MAX_BATCH_SIZE变量。
摘要由CSDN通过智能技术生成

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
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值