多卡运行

[song-leap]

Hello！

May I ask if this project can run on a multi-GPU environment? I have tried to add code for multi-GPU running in my code multiple times, such as os.environCUDA_VISIBLE_DEVICES= '0,3', torch.nn.DataParallel(model,device_ids=[0,3]), etc. However, the system always throws out error messages. I'm wondering if the project code is set to run only in a single GPU environment?

Best,
Nan