Revert "Removing CUDA deps"

This reverts commit d52b754794230eaaeac972be5f9341298f50ce30.

requirements.txt

@@ -162,6 +162,19 @@ ninja==1.11.1.4
 nltk==3.9.1
 numexpr==2.10.2
 numpy==2.1.2
+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-ml-py==12.570.86
+nvidia-nccl-cu12==2.21.5
+nvidia-nvjitlink-cu12==12.1.105
+nvidia-nvtx-cu12==12.1.105
 oauthlib==3.2.2
 objprint==0.3.0
 oci==2.150.0
@@ -291,7 +304,7 @@ tld==0.13
 tldextract==5.2.0
 tokenizers==0.21.1
 toolz==1.0.0
-torch==2.5.0
+torch==2.5.0+cu121
 tornado==6.4.2
 tqdm==4.67.1
 tqdm-multiprocess==0.0.11