ModelScope中rex 推理的时候能在指定的cuda 上进行推理吗? 能传cudaid吗
在ModelScope中,您可以指定哪些GPU应该被用于计算。这可以通过设置CUDA_VISIBLE_DEVICES环境变量来实现。具体的操作步骤是:打开终端,输入export CUDA_VISIBLE_DEVICES=0
,其中的数字“0”代表的是指定的GPU设备编号。通过这样的方式,您可以选择在特定的cuda上进行推理。
在ModelScope中,您可以选择使用CUDA加速或CPU运行来进行文本处理任务。对于具体的配置方式,您可以通过设置环境变量来指定使用的CUDA设备。例如,如果您希望使用第一个GPU设备,可以在运行命令时添加以下环境变量:
export CUDA_VISIBLE_DEVICES=0
通过script运行可以,再加一下os.environ['CUDA_VISIBLE_DEVICES']='0' ,此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”