ModelScope训练的时候,能指定不同显卡申请的显存大小吗?
ModelScope 训练的时候,可以通过 -gpu_list 参数指定 GPU 列表,以及通过 -gpu_assign_mem_mb 参数指定不同 GPU 申请显存的大小。
例如,以下命令指定了两个 GPU,第一个 GPU 申请 12G 显存,第二个 GPU 申请 8G 显存:
python3 modelscope train --dataset_type imagenet --model_type resnet34 --gpu_list 0,1 --gpu_assign_mem_mb 1