Fix requirements

TUM-DAML · May 30, 2022 · 0110dce · 0110dce
1 parent 1922774
commit 0110dce
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/env.yml b/env.yml
@@ -6,11 +6,9 @@ channels:
   - pyg
 dependencies:
   - python==3.8
-  - pytorch==1.10
-  - torchvision 
   - cudatoolkit=11.3
-  - torchaudio 
-  - pytorch-geometric
+  - pytorch==1.10
+  - pytorch-scatter
   - jupyterlab
   - numpy
   - numba

diff --git a/requirements.txt b/requirements.txt
@@ -6,5 +6,5 @@ ase
 nglview
 numba
 jupyterlab
-torchvision 
-torchaudio 
+torch==1.10
+torch-scatter
diff --git a/setup.py b/setup.py
@@ -12,4 +12,5 @@
     packages=["gemnet"],
     install_requires=install_requires,
     zip_safe=False,
+    python_requires='>=3.8',
 )