ci: try to tie cuda version to scatter

GT4SD · Mar 7, 2024 · 175cc89 · 175cc89
1 parent c0273b8
commit 175cc89
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/conda.yml b/conda.yml
@@ -9,7 +9,7 @@ dependencies:
   - pip>=19.1,<20.3
   - pytorch>=1.0,<=1.12.1
   - cpuonly
-  - pytorch-scatter<=2.0.9
+  - pytorch-scatter<=2.0.9,cudatoolkit=10.2
   - pip:
       - -r requirements.txt
       - -r cpu_requirements.txt

diff --git a/conda_gpu.yml b/conda_gpu.yml
@@ -8,7 +8,7 @@ dependencies:
   - python>=3.7,<3.9
   - pip>=19.1,<20.3
   - pytorch>=1.0,<=1.12.1=*cu*
-  - pytorch-scatter<=2.0.9
+  - pytorch-scatter<=2.0.9,cudatoolkit=10.2
   - torchvision<=0.13.1=*cu*
   - torchaudio<=0.12.1=*cu*
   - pip: