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

请问ModelScope Qwen的推理使用GPU如何制定device id ?

请问ModelScope Qwen的推理使用GPU如何制定device id ? c90c3568c1a6713722307667f3980990.png
69fbefb0de7cf0c0d7abf2c8ceb71383.png

展开
收起
小小爱吃香菜 2024-02-13 21:14:38 111 0
1 条回答
写回答
取消 提交回答
  • model.from_pretrained指定device="cuda:0"。可能不需要device_map。 此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2024-02-14 13:08:53
    赞同 展开评论 打赏

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

热门讨论

热门文章

相关电子书

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