Update requirements.txt

requirements.txt

@@ -95,18 +95,6 @@ networkx==3.2.1
 notebook==7.3.2
 notebook_shim==0.2.4
 numpy==1.26.3
-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.1.105
-nvidia-nvtx-cu12==12.1.105
 overrides==7.7.0
 packaging==24.2
 pandas==2.2.3
@@ -160,9 +148,9 @@ timeout-decorator==0.5.0
 tinycss2==1.4.0
 toml==0.10.2
 tomli==2.2.1
-torch==2.4.1+cu121
-torchaudio==2.4.1+cu121
-torchvision==0.19.1+cu121
+torch==2.4.1
+torchaudio==2.4.1
+torchvision==0.19.1
 tornado==6.4.2
 tqdm==4.67.1
 traitlets==5.14.3