fix installation configuration

BUPT-GAMMA · Dec 19, 2024 · a41bba4 · a41bba4
1 parent 1a8977b
commit a41bba4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -142,7 +142,7 @@ Currently, GammaGL requires **Python Version >= 3.9** and is only supported on *
 
    # For torch, version 2.1+cuda 11.8
    # https://pytorch.org/get-started/locally/
-   $ pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+   $ pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118
 
    # For paddle, any latest stable version
    # https://www.paddlepaddle.org.cn/

diff --git a/setup.py b/setup.py
@@ -118,8 +118,8 @@ def load_extensions():
 
     return extensions
 
-install_requires = ['numpy==1.24', 'pandas', 'numba', 'scipy', 'protobuf', 'pyparsing', 'rdkit',
-                    'tensorboardx', 'pytest', 'tensorlayerx', 'rich', 'tqdm', 'pybind11', 'panda', 'ninja']
+install_requires = ['numpy==1.24', 'pandas', 'numba==0.59.0', 'scipy', 'protobuf', 'pyparsing', 'rdkit',
+                    'tensorboardx', 'pytest', 'tensorlayerx', 'rich', 'tqdm', 'pybind11', 'panda', 'ninja==1.11.1.1']
 
 classifiers = [
     'Development Status :: 3 - Alpha',