update torch and torch vision for mnist test

Author: kryanbeane

tests/e2e/mnist_pip_requirements.txt

@@ -1,6 +1,6 @@
 --extra-index-url https://download.pytorch.org/whl/cu118
 torch==2.7.1+cu118
-torchvision==0.20.1
+torchvision==0.22.1+cu118
 pytorch_lightning==1.9.5
 torchmetrics==0.9.1
 minio