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

ModelScope这个模型不支持GPU多卡吗?这里设置2,但是从nVidia-smi上看,只有一张

问题1:https://modelscope.cn/models/damo/nlp_csanmt_translation_en2zh/summary ModelScope这个模型不支持GPU多卡吗?2ad90c3c217fced85296ffedc3f98bc0.png 这里设置2,但是从nVidia-smi上看,只有一张卡在跑。是不是因为使用的是导出的savedmodel? 问题2:352575920baf74b22566822a3d84f734.png 这个参数仅限于train时吧?推导时不能用多卡?

展开
收起
cuicuicuic 2023-06-04 20:30:58 246 0
1 条回答
写回答
取消 提交回答
  • 回答1:从代码上看似乎是支持多GPU的。 这个模型是支持多GPU的,应该是基于tf_ckpts目录下的checkpoint跑,num_gpus=2指的就是2卡跑任务。 回答2:多卡仅限于训练,不支持多卡推理,此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2023-06-04 20:46:51
    赞同 展开评论 打赏

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

相关电子书

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