AI Cube是否支持50系显卡训练,训练卡死在第一轮

Viewed 73

重现步骤

  1. 5070Ti 在AI Cube上进行默认配置的目标检测模型训练

期待结果和实际结果

期待结果:正常训练输出
实际结果:始终卡死在第一轮

软硬件版本信息

AI Cube 1.40
5070Ti显卡

错误日志

在相关log中未见到相关报错日志

尝试解决过程

我认为是pytorch环境相关引起的问题,因为50系的cuda环境比较高,是不支持pytorch一些早期版本的,而AI Cube又是内置pytorch环境,我估计没有考虑这么多

补充材料

1 Answers

没有试过5070Ti显卡

我怀疑是pytorch环境的问题,因为50系显卡的cuda版本很高,是不支持前几个版本的pytorch的,而AI Cube是内置pytorch环境,我看了一下好像版本太低,但是不知道怎么更换

有可能,但是AICube的torch是打包到安装目录中的,不好换

你如果有训练的需求,可以使用yolo那一套