ModelScope训练的时候,能指定不同显卡申请的显存大小吗?

ModelScope训练的时候,能指定不同显卡申请的显存大小吗?

展开
收起
青城山下庄文杰 2023-08-05 20:29:22 180 分享 版权
1 条回答
写回答
取消 提交回答
  • ModelScope 训练的时候,可以通过 -gpu_list 参数指定 GPU 列表,以及通过 -gpu_assign_mem_mb 参数指定不同 GPU 申请显存的大小。

    例如,以下命令指定了两个 GPU,第一个 GPU 申请 12G 显存,第二个 GPU 申请 8G 显存:

    python3 modelscope train --dataset_type imagenet --model_type resnet34 --gpu_list 0,1 --gpu_assign_mem_mb 1

    2023-09-26 11:17:05
    赞同 展开评论

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

热门讨论

热门文章

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