Since project contains a lot of files, creating small snippet seems a difficult task to me. Can to suggest me how to debug the code for multi-gpu support ?
Can you think of any other reason why DataParallel not working based on your experience with pytorch and cuda ?
@ptrblck Thanks a lot for taking time to help me out