设置multi-gpu实际只挂载一个

[carol007]

你好，我想加速imagenet的读写，并修改了你的代码 从 train_loader = get_imagenet_iter_dali(type='train', image_dir='/userhome/memory_data/imagenet', batch_size=256, num_threads=4, crop=224, device_id=0, num_gpus=1) 修改至 train_loader = get_imagenet_iter_dali(type='train', image_dir='/data1/share', batch_size=128, num_threads=4, crop=224,device_id=(0,1,2,3), num_gpus=4)

但是CPU占用率相当高85%。且实际只有 gpu = 0被挂载，1，2，3没有。0的util使用率只有5%

[SonpKing]

因为作者根本没实现多GPU

[One-punch24]

the problem seems to be here? I am not sure, and want somebody to teach me where is wrong?