将ModelScope mgeo模型放到gpu上了也按照官网上说的做了但最后还是用的cpu该如何解决

将ModelScope mgeo模型放到gpu上了,也按照官网上说的做了,但最后还是用的cpu,该如何解决?

展开
收起
2401。 2023-09-02 20:44:08 109 分享 版权
1 条回答
写回答
取消 提交回答
  • python执行代码前面加上CUDA_VISIBLE_DEVICES=0 试试呢。
    此回答整理自“魔搭ModelScope开发者联盟群 ①”。

    2023-09-05 12:20:15
    赞同 展开评论

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

热门讨论

热门文章

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