Upgrade PyTorch version to v2.2.1 #1102

Job	Run time
Publish Image (darts-cnn-cifar10-cpu, linux/amd64,linux/arm64, examples/v1beta1/trial-images/dart... / Publish Image	8m 5s
Publish Image (darts-cnn-cifar10-gpu, linux/amd64, examples/v1beta1/trial-images/darts-cnn-cifar1... / Publish Image	5m 42s
Publish Image (enas-cnn-cifar10-cpu, linux/amd64,linux/arm64, examples/v1beta1/trial-images/enas-... / Publish Image	6m 50s
Publish Image (enas-cnn-cifar10-gpu, linux/amd64, examples/v1beta1/trial-images/enas-cnn-cifar10/... / Publish Image	2m 57s
Publish Image (pytorch-mnist-cpu, linux/amd64,linux/arm64, examples/v1beta1/trial-images/pytorch-... / Publish Image	7m 59s
Publish Image (pytorch-mnist-gpu, linux/amd64, examples/v1beta1/trial-images/pytorch-mnist/Docker... / Publish Image	5m 23s
Publish Image (simple-pbt, linux/amd64,linux/arm64, examples/v1beta1/trial-images/simple-pbt/Dock... / Publish Image	1m 12s
Publish Image (tf-mnist-with-summaries, linux/amd64,linux/arm64, examples/v1beta1/trial-images/tf... / Publish Image	5m 59s
	44m 7s

Provide feedback