开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

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

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

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

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

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

相关电子书

更多
DeepStream: GPU加速海量视频数据智能处理 立即下载
阿里巴巴高性能GPU架构与应用 立即下载
GPU在超大规模深度学习中的发展和应用 立即下载