update torch

requirements.txt

@@ -25,17 +25,28 @@ mdurl==0.1.2
 mpmath==1.3.0
 networkx==3.4.2
 numpy==2.2.0
+nvidia-cublas-cu11==11.11.3.6
 nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu11==11.8.87
 nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu11==11.8.89
 nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu11==11.8.89
 nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu11==9.1.0.70
 nvidia-cudnn-cu12==9.1.0.70
+nvidia-cufft-cu11==10.9.0.58
 nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu11==10.3.0.86
 nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu11==11.4.1.48
 nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu11==11.7.5.86
 nvidia-cusparse-cu12==12.1.0.106
+nvidia-nccl-cu11==2.20.5
 nvidia-nccl-cu12==2.20.5
 nvidia-nvjitlink-cu12==12.6.85
+nvidia-nvtx-cu11==11.8.86
 nvidia-nvtx-cu12==12.1.105
 orjson==3.10.12
 packaging==24.2
@@ -68,7 +79,9 @@ starlette==0.41.3
 sympy==1.13.3
 tokenizers==0.21.0
 tomlkit==0.13.2
-torch==2.4.1
+torch==2.4.0+cu118
+torchaudio==2.4.0+cu118
+torchvision==0.19.0+cu118
 tqdm==4.67.1
 transformers==4.47.0
 triton==3.0.0