Solved by Ahtsham_Manzoor in post #3 
              
              
                I solved the issue by upgrading the Pytorch from 1.5.0 to 1.6.0 with Cuda 10.1. Also, in the command, I explicitly mentioned the GPU_id=0 as I have just one GPU in my device.