开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

ModelScope中rex 推理的时候能在指定的cuda 上进行推理吗?

ModelScope中rex 推理的时候能在指定的cuda 上进行推理吗? 能传cudaid吗image.png

展开
收起
真的很搞笑 2024-01-02 20:11:59 47 1
3 条回答
写回答
取消 提交回答
  • 面对过去,不要迷离;面对未来,不必彷徨;活在今天,你只要把自己完全展示给别人看。

    在ModelScope中,您可以指定哪些GPU应该被用于计算。这可以通过设置CUDA_VISIBLE_DEVICES环境变量来实现。具体的操作步骤是:打开终端,输入export CUDA_VISIBLE_DEVICES=0,其中的数字“0”代表的是指定的GPU设备编号。通过这样的方式,您可以选择在特定的cuda上进行推理。

    2024-01-03 13:48:47
    赞同 展开评论 打赏
  • 在ModelScope中,您可以选择使用CUDA加速或CPU运行来进行文本处理任务。对于具体的配置方式,您可以通过设置环境变量来指定使用的CUDA设备。例如,如果您希望使用第一个GPU设备,可以在运行命令时添加以下环境变量:

    export CUDA_VISIBLE_DEVICES=0
    
    2024-01-03 09:49:28
    赞同 展开评论 打赏
  • 通过script运行可以,再加一下os.environ['CUDA_VISIBLE_DEVICES']='0' ,此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2024-01-02 22:36:22
    赞同 展开评论 打赏

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

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
低代码开发师(初级)实战教程 立即下载