GCP/A3 doesn't work with torch<2.2 - suggesting to change the deps

GCP/A3 doesn't work with torch<2.2 - suggesting to change the deps #2163