require torch 1.6

biandh · Mar 21, 2021 · e4cdab9 · e4cdab9
1 parent 9c28d0e
commit e4cdab9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -41,16 +41,14 @@ The release note is available [here](https://github.com/snap-stanford/ogb/releas
 
 #### Requirements
  - Python>=3.6
- - PyTorch>=1.2
+ - PyTorch>=1.6
  - DGL>=0.5.0 or torch-geometric>=1.6.0
  - Numpy>=1.16.0
  - pandas>=0.24.0
  - urllib3>=1.24.0
  - scikit-learn>=0.20.0
  - outdated>=0.2.0
 
- **Note:** `torch-geometric>=1.6.0` is recommended to run our [example code](https://github.com/snap-stanford/ogb/tree/master/examples).
-
 #### Pip install
 The recommended way to install OGB is using Python's package manager pip:
 ```bash

diff --git a/setup.py b/setup.py
@@ -27,7 +27,7 @@
       long_description=long_description,
       long_description_content_type='text/markdown',
       install_requires = [
-        'torch>=1.2.0',
+        'torch>=1.6.0',
         'numpy>=1.16.0',
         'tqdm>=4.29.0',
         'scikit-learn>=0.20.0',