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

为什么用ModelScope数据集可以训练,但是安装后为啥显示CPU版本呢?"

"为什么用ModelScope
https://modelscope.cn/datasets/AI-ModelScope/blossom-math-v2
数据集可以训练,
但是用
https://modelscope.cn/datasets/gavinluo/problem
训练就会报错,安装的是CUDA11.7pytorch2.02,但是安装后为啥显示CPU版本呢?"

展开
收起
小小爱吃香菜 2024-08-06 21:59:59 42 0
1 条回答
写回答
取消 提交回答
  • "problem不在支持的数据集列表中,您看下
    nvidia-smi
    nvcc --version 您重装一下torch叭。此外,如果没特殊要求的话,cuda可以装12.1版本的。 此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”"

    2024-08-06 22:47:05
    赞同 1 展开评论 打赏

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

相关电子书

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