Skip to content

Commit

Permalink
Merge pull request #192 from jamesmyatt/update-installation
Browse files Browse the repository at this point in the history
Update installation instructions for PyTorch 1.11.0 and PyG latest
  • Loading branch information
benedekrozemberczki committed Aug 30, 2022
2 parents c5ed919 + f2b3baf commit db1dce2
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 34 deletions.
13 changes: 4 additions & 9 deletions .github/workflows/main.yml
Original file line number Diff line number Diff line change
@@ -1,21 +1,19 @@
name: CI

on:

push:
branches: [ master ]
pull_request:
branches: [ master ]

workflow_dispatch:

jobs:
build:
runs-on: ${{ matrix.os }}

strategy:
matrix:
os: [ubuntu-18.04]
os: [ubuntu-20.04]

steps:
- uses: actions/checkout@v2
Expand All @@ -30,11 +28,8 @@ jobs:
- run: which python
- name: Install main dependencies
run: |
conda install -y scipy
python -m pip install torch==1.10.1+cpu torchvision==0.11.2+cpu torchaudio==0.10.1+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
python -m pip install torch-scatter -f https://data.pyg.org/whl/torch-1.10.0+cpu.html
python -m pip install torch-sparse -f https://data.pyg.org/whl/torch-1.10.0+cpu.html
python -m pip install torch-geometric
python -m pip install torch==1.11.0
python -m pip install torch-scatter torch-sparse torch-cluster torch-geometric -f https://data.pyg.org/whl/torch-1.11.0+cpu.html
python -m pip install sphinx sphinx_rtd_theme
- name: Install main package
run: |
Expand Down
19 changes: 4 additions & 15 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -163,26 +163,15 @@ If you notice anything unexpected, please open an [issue](https://benedekrozembe

**Installation**

Binaries are provided for Python version <= 3.9.

**PyTorch 1.10.0**

To install the binaries for PyTorch 1.10.0, simply run
First install [pytorch][pytorch-install] and [pytorch-geometric][pyg-install]
and then run

```sh
pip install torch-scatter -f https://data.pyg.org/whl/torch-1.10.0+${CUDA}.html
pip install torch-sparse -f https://data.pyg.org/whl/torch-1.10.0+${CUDA}.html
pip install torch-geometric
pip install torch-geometric-temporal
```

where `${CUDA}` should be replaced by either `cpu`, `cu102`, or `cu113` depending on your PyTorch installation.

| | `cpu` | `cu102` | `cu113` |
|-------------|-------|---------|---------|
| **Linux** ||||
| **Windows** ||||
| **macOS** || | |
[pytorch-install]: https://pytorch.org/get-started/locally/
[pyg-install]: https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html

--------------------------------------------------------------------------------

Expand Down
11 changes: 1 addition & 10 deletions docs/source/notes/installation.rst
Original file line number Diff line number Diff line change
Expand Up @@ -2,22 +2,13 @@ Installation
============

The installation of PyTorch Geometric Temporal requires the presence of certain prerequisites. These are described in great detail in the installation description of PyTorch Geometric. Please follow the instructions laid out `here <https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html>`_. You might also take a look at the `readme file <https://github.com/benedekrozemberczki/pytorch_geometric_temporal>`_ of the PyTorch Geometric Temporal repository.
Binaries are provided for Python version <= 3.9.

**PyTorch 1.10.0**

To install the binaries for PyTorch 1.10.0, simply run
Once the required versions of PyTorch and PyTorch Geometric are installed, simply run:

.. code-block:: none
$ pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.10.0+${CUDA}.html
$ pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.10.0+${CUDA}.html
$ pip install torch-geometric
$ pip install torch-geometric-temporal
where `${CUDA}` should be replaced by either `cpu`, `cu102`, or `cu113` depending on your PyTorch installation.

**Updating the Library**

The package itself can be installed via pip:
Expand Down

0 comments on commit db1dce2

Please sign in to comment.