windows版AI Cube怎么配置用gpu训练

Question

重现步骤
AI Cube V1.4 for Windows训练速度慢

期待结果和实际结果
怎么配置gpu训练模型，已经训练20个小时了

软硬件版本信息
win10 64位专业版 32g内存 4070s显卡，训练用的默认配置参数
AI Cube V1.4 for Windows

错误日志

尝试解决过程

补充材料

Wy001 · Answer

AICube不支持CPU训练，默认使用的就是GPU，按照用户指南操作即可。如果训练慢，请具体说明怎么慢，什么任务，多少数据，batchsize设置的多少。同时训练启动时，注意查看nvidia-smi下的GPU占用信息，是否有显存消耗。