diff --git a/conda.yml b/conda.yml index e1fcb522a..0c7887c2e 100644 --- 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 index 47c27265a..e40727673 100644 --- 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: