问题1:https://modelscope.cn/models/damo/nlp_csanmt_translation_en2zh/summary ModelScope这个模型不支持GPU多卡吗? 这里设置2,但是从nVidia-smi上看,只有一张卡在跑。是不是因为使用的是导出的savedmodel? 问题2:
这个参数仅限于train时吧?推导时不能用多卡?
回答1:从代码上看似乎是支持多GPU的。 这个模型是支持多GPU的,应该是基于tf_ckpts目录下的checkpoint跑,num_gpus=2指的就是2卡跑任务。 回答2:多卡仅限于训练,不支持多卡推理,此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”
ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!欢迎加入技术交流群:微信公众号:魔搭ModelScope社区,钉钉群号:44837352