ModelScope训练的时候怎么设置显卡?

ModelScope训练的时候怎么设置显卡?

展开
收起
真的很搞笑 2023-10-19 12:30:04 402 分享 版权
2 条回答
写回答
取消 提交回答
  • 面对过去,不要迷离;面对未来,不必彷徨;活在今天,你只要把自己完全展示给别人看。

    根据提供的信息,您需要在ModelScope平台上设置训练任务使用的显卡。具体来说,您可以在设置训练任务时,选择要使用的GPU。在设置GPU时,您需要提供GPU的ID。在ModelScope平台上,您可以使用CUDA_VISIBLE_DEVICES环境变量来设置要使用的GPU。例如,如果您想要使用GPU0和GPU1进行训练,可以在运行训练任务时添加以下命令:

    CUDA_VISIBLE_DEVICES=0,1 python train.py
    

    在上述命令中,CUDA_VISIBLE_DEVICES=0,1表示要使用GPU0和GPU1进行训练。建议您在运行训练任务时,检查GPU的ID是否正确,以确保训练任务能够正确使用显卡。

    2023-10-19 14:00:46
    赞同 展开评论
  • sh脚本中有设置这个参数0532e249b12450fd8f5d2df2a38c2634.png
    ,此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2023-10-19 12:40:09
    赞同 展开评论

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

热门讨论

热门文章

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