**Is your feature request related to a problem? Please describe.** pytorch 1.10 is available, would be great to include it in the testing pipelines.