开发者社区 > ModelScope模型即服务 > 正文

ModelScope为啥使用notebook的cpu版本,finetune.sh脚本去训练一直失败?

为啥使用notebook的cpu版本,运行examples/industrial_data_pretraining里面的finetune.sh脚本去训练一直失败,我明明添加了device参数为cpu了,但是根据输出日志,它还是会去寻找cuda。而且同样的脚本使用notebook的gpu版本来运行就一点问题都没有了f090c12adf37037dabb3ba2abea9e478.png

展开
收起
防火防爆 2024-08-26 14:13:27 40 0
1 条回答
写回答
取消 提交回答
  • 3f5ee7a881e3d5c75c4bab01a04cd3d2.png这里debug看下哈不过一般训练用gpu 此回答整理自钉钉群“魔搭ModelScope开发者联盟群 ①”

    2024-08-28 13:13:07
    赞同 展开评论 打赏

ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!欢迎加入技术交流群:微信公众号:魔搭ModelScope社区,钉钉群号:44837352

热门讨论

热门文章

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
RDS SQL Server CPU高使用率性能优化 立即下载