update dependency to torch 1.8.0

DeepGraphLearning · Nov 21, 2021 · a710097 · a710097
1 parent 909dd2d
commit a710097
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 10 deletions.
diff --git a/README.md b/README.md
@@ -26,12 +26,12 @@ Installation
 ------------
 
 TorchDrug can be installed on either Linux, Windows or macOS. It is compatible with
-Python 3.7/3.8 and PyTorch >= 1.4.0.
+Python 3.7/3.8 and PyTorch >= 1.8.0.
 
 ### From Conda ###
 
 ```bash
-conda install -c milagraph -c conda-forge torchdrug
+conda install torchdrug -c milagraph -c conda-forge -c pytorch -c pyg
 ```
 
 ### From Pip ###

diff --git a/conda/torchdrug/meta.yaml b/conda/torchdrug/meta.yaml
@@ -11,8 +11,8 @@ requirements:
     - pip
   run:
     - python >=3.7,<3.9
-    - pytorch >=1.4.0
-    - pytorch_scatter >=1.4.0
+    - pytorch >=1.8.0
+    - pytorch-scatter >=2.0.8
     - decorator
     - numpy >=1.11
     - rdkit

diff --git a/doc/source/installation.rst b/doc/source/installation.rst
@@ -2,7 +2,7 @@ Installation
 ============
 
 TorchDrug can be installed on either Linux, Windows or macOS. It is compatible with
-Python 3.7/3.8 and PyTorch >= 1.4.0.
+Python 3.7/3.8 and PyTorch >= 1.8.0.
 
 For Windows
 
@@ -11,7 +11,7 @@ From Conda
 
 .. code:: bash
 
-    conda install -c milagraph -c conda-forge torchdrug
+    conda install torchdrug -c milagraph -c conda-forge -c pytorch -c pyg
 
 From Pip
 --------

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
-torch>=1.4.0
-torch-scatter>=1.4.0
+torch>=1.8.0
+torch-scatter>=2.0.8
 decorator
 numpy>=1.11
 rdkit-pypi

diff --git a/setup.py b/setup.py
@@ -29,8 +29,8 @@
         test_suite="nose.collector",
         install_requires=
             [
-                "torch>=1.4.0",
-                "torch-scatter>=1.4.0",
+                "torch>=1.8.0",
+                "torch-scatter>=2.0.8",
                 "decorator",
                 "numpy>=1.11",
                 "rdkit-pypi",