为什么用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 80 分享 版权
1 条回答
写回答
取消 提交回答
  • "problem不在支持的数据集列表中,您看下
    nvidia-smi
    nvcc --version 您重装一下torch叭。此外,如果没特殊要求的话,cuda可以装12.1版本的。 此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”"

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

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

还有其他疑问?
咨询AI助理