requirements update

requirements.txt

@@ -34,6 +34,18 @@ mdurl==0.1.2
 mpmath==1.3.0
 networkx==3.2.1
 numpy==2.1.1
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu12==9.1.0.70
+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-nvjitlink-cu12==12.6.68
+nvidia-nvtx-cu12==12.1.105
 omegaconf==2.3.0
 opencv-python==4.10.0.84
 orjson==3.10.7
@@ -64,11 +76,12 @@ sympy==1.12
 timm==1.0.9
 tokenizers==0.19.1
 tomlkit==0.12.0
-torch==2.4.1+cpu
-torchaudio==2.4.1+cpu
-torchvision==0.19.1+cpu
+torch==2.4.1
+torchaudio==2.4.1
+torchvision==0.19.1
 tqdm==4.66.5
 transformers==4.44.2
+triton==3.0.0
 typer==0.12.5
 typing_extensions==4.12.2
 tzdata==2024.1