How to create vendor agnostic docker Pytorch image for Training in GCP tesla GPU and Prediction in Azure tesla GPU ?