get torch 2.2.0 working

requirements.txt

@@ -38,7 +38,7 @@ nvidia-cufft-cu12==11.0.2.54
 nvidia-curand-cu12==10.3.2.106
 nvidia-cusolver-cu12==11.4.5.107
 nvidia-cusparse-cu12==12.1.0.106
-nvidia-nccl-cu12==2.20.5
+nvidia-nccl-cu12==2.19.3
 nvidia-nvjitlink-cu12==12.5.40
 nvidia-nvtx-cu12==12.1.105
 pandas==2.2.2
@@ -60,8 +60,8 @@ setproctitle==1.3.3
 smmap==5.0.1
 sympy==1.12.1
 tokenizers==0.19.1
-torch==2.3.0
-torchvision==0.18.0
+torch==2.2.0
+torchvision==0.17.0
 tqdm==4.66.4
 transformers==4.41.1
 typing-extensions==4.12.0