开发者社区 > ModelScope模型即服务 > 正文

麻烦问一下ModelScope,配置train.yaml怎么指定gpu?在文档里并没有写明怎么指定g

麻烦问一下ModelScope,配置train.yaml怎么指定gpu?在文档里并没有写明怎么指定gpu号?

展开
收起
真的很搞笑 2023-04-22 14:14:07 478 0
1 条回答
写回答
取消 提交回答
  • 您好,在train.yaml文件中,您可以通过以下方式指定GPU:

    train:
    gpus: [0, 1] # 指定要使用的GPU编号,多个GPU之间用逗号分隔
    如果您想使用所有可用的GPU,也可以使用以下配置:

    train:
    gpus: -1 # 使用所有可用的GPU
    希望这些信息对您有所帮助。如果您还有其他问题,请随时告诉我。

    2024-02-28 17:35:21
    赞同 展开评论 打赏

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

相关电子书

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