NaSGEC icon indicating copy to clipboard operation
NaSGEC copied to clipboard

mutiple gpus

Open leleP9 opened this issue 1 year ago • 4 comments

使用单卡执行generate.sh时显存不够,在设置CUDA_DEVICE=0,1后执行仍使用单卡预测,请问该怎么实现多卡预测?

leleP9 avatar Jan 18 '24 06:01 leleP9