根据提供的信息,您需要在ModelScope平台上设置训练任务使用的显卡。具体来说,您可以在设置训练任务时,选择要使用的GPU。在设置GPU时,您需要提供GPU的ID。在ModelScope平台上,您可以使用CUDA_VISIBLE_DEVICES环境变量来设置要使用的GPU。例如,如果您想要使用GPU0和GPU1进行训练,可以在运行训练任务时添加以下命令:
CUDA_VISIBLE_DEVICES=0,1 python train.py
在上述命令中,CUDA_VISIBLE_DEVICES=0,1表示要使用GPU0和GPU1进行训练。建议您在运行训练任务时,检查GPU的ID是否正确,以确保训练任务能够正确使用显卡。