fix: removed nvidia requirements (only test on cpu)

recursionpharma · Feb 16, 2024 · b10dfa4 · b10dfa4
1 parent 5c15ede
commit b10dfa4
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 24 deletions.
diff --git a/requirements/dev-3.10.txt b/requirements/dev-3.10.txt
@@ -50,18 +50,6 @@ networkx==3.2.1
 nodeenv==1.8.0 
 numexpr==2.9.0 
 numpy==1.26.4 
-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==8.9.2.26 
-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.18.1 
-nvidia-nvjitlink-cu12==12.3.101 
-nvidia-nvtx-cu12==12.1.105 
 omegaconf==2.3.0 
 opt-einsum==3.3.0 
 packaging==23.2 

diff --git a/requirements/main-3.10.txt b/requirements/main-3.10.txt
@@ -35,18 +35,6 @@ ndindex==1.8
 networkx==3.2.1 
 numexpr==2.9.0 
 numpy==1.26.4 
-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==8.9.2.26 
-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.18.1 
-nvidia-nvjitlink-cu12==12.3.101 
-nvidia-nvtx-cu12==12.1.105 
 omegaconf==2.3.0 
 opt-einsum==3.3.0 
 packaging==23.2