Implement MultiGPU in Pytorch

Job ID: 34869865

Budget: ₹600 – ₹1,500 INR

I have a piece of code that has support for MultiGPU but the code is implemented only for GPU[0] explicitly. I want to you help me change the code such that all 4 GPUs are used.

Related categories: Algorithm CUDA Machine Learning (ML) Tensorflow Pytorch