Running load using official Nvidia PyTorch image boost performance by 50%