CUDA 10.2 and pytorch 1.7.1 support needed - Runpod