Skip to content

Commit

Permalink
Update env_setup to be compatible with pyproject with torch
Browse files Browse the repository at this point in the history
  • Loading branch information
luisfpereira committed Nov 22, 2024
1 parent 804d15b commit 6c71105
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions env_setup.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,4 @@
#!/bin/bash -l
pip install --upgrade pip
pip install -e '.[all]'

# Note that not all combinations of torch and CUDA are available
# See https://github.com/pyg-team/pyg-lib to check the configuration that works for you
Expand All @@ -10,6 +8,7 @@ CUDA="cu118" # if available, select the CUDA version suitable for your system
pip install torch==${TORCH} --extra-index-url https://download.pytorch.org/whl/${CUDA}
pip install pyg-lib torch-scatter torch-sparse torch-cluster -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html

#pytest
pip install --upgrade pip
pip install -e '.[all]'

pre-commit install

0 comments on commit 6c71105

Please sign in to comment.