Skip to content

Commit 25f77c2

Browse files
committed
Update PopTorch compatibility info
Summary: Update compatibility tables. Remove previous installation workaround !docs-only !ci_branch_mk2 Reviewers: grahamh, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved, #pytorch, #tech_docs, jacekr, dariuszs Reviewed By: #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved, #pytorch, #tech_docs, jacekr, dariuszs Subscribers: jacekr JIRA Issues: DOCS-77 Differential Revision: https://phabricator.sourcevertex.net/D84779
1 parent 8e04111 commit 25f77c2

2 files changed

Lines changed: 8 additions & 16 deletions

File tree

docs/poptorch_geometric/user_guide/installation.rst

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,12 @@ versions and supported Python versions.
2828
+------------------------+-----------------------+-------------+-----------------+----------------+------------+
2929
| ``poptorch_geometric`` | ``pytorch_geometric`` | ``torch`` | ``torchvision`` | ``torchaudio`` | ``python`` |
3030
+========================+=======================+=============+=================+================+============+
31+
| 3.3 | 2.4.0.dev20230613 | 2.0.1 | 0.15.2 | 2.0.1 | >= 3.8 |
32+
+------------------------+-----------------------+-------------+-----------------+----------------+------------+
3133
| 3.2 | 2.3.0.dev20230222 | 1.13.1 | 0.14.1 | 0.13.1 | >= 3.7 |
3234
+------------------------+-----------------------+-------------+-----------------+----------------+------------+
3335

36+
.. note:: To ensure version compatibility, ``torchvision`` and ``torchaudio`` are automatically installed with PopTorch in Poplar SDK 3.3 and later.
3437

3538
Installation using Python virtual environment
3639
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@@ -50,14 +53,6 @@ After creating the virtual environment, you need to install the PopTorch wheel.
5053
5154
$ pip install <sdk_path>/poptorch-x.x.x.whl
5255
53-
.. warning:: If, after installing PopTorch, you install a third-party library that requires ``torchvision`` or ``torchaudio`` then that may cause an incompatible version of ``torch`` to be installed.
54-
55-
To prevent this, after installing PopTorch, use the following commands to install compatible versions of ``torchvision`` or ``torchaudio``:
56-
57-
.. code-block:: bash
58-
59-
$ pip install torchvision==0.14.1+cpu --extra-index-url https://download.pytorch.org/whl/cpu
60-
$ pip install torchaudio==0.13.1+cpu --extra-index-url https://download.pytorch.org/whl/cpu
6156
6257
See the
6358
`PopTorch installation guide <https://docs.graphcore.ai/projects/poptorch-user-guide/en/latest/installation.html>`_

docs/user_guide/installation.rst

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,8 @@ The following are the corresponding ``torch``, ``torchvision``, ``torchaudio`` a
2929
+--------------+-----------+-----------------+----------------+------------------------------+------------+
3030
| ``poptorch`` | ``torch`` | ``torchvision`` | ``torchaudio`` | ``torch_scatter`` | ``python`` |
3131
+==============+===========+=================+================+==============================+============+
32+
| 3.3 | 2.0.1 | 0.15.2 | 2.0.1 | >=2.0.9 and <=2.1.1 | >=3.8 |
33+
+--------------+-----------+-----------------+----------------+------------------------------+------------+
3234
| 3.2 | 1.13.1 | 0.14.1 | 0.13.1 | >=2.0.9 and <=2.1.0 | >=3.7 |
3335
+--------------+-----------+-----------------+----------------+------------------------------+------------+
3436
| 3.1 | 1.13.0 | 0.14.0 | 0.13.0 | >=2.0.9 and <=2.1.0 | >=3.7 |
@@ -54,6 +56,9 @@ The following are the corresponding ``torch``, ``torchvision``, ``torchaudio`` a
5456

5557
Based on https://github.com/pytorch/vision/blob/master/README.md
5658

59+
.. note:: To ensure version compatibility, ``torchvision`` and ``torchaudio`` are automatically installed with PopTorch in Poplar SDK 3.3 and later.
60+
61+
5762
Using a Python virtual environment
5863
==================================
5964

@@ -68,14 +73,6 @@ for this. You can create a virtual environment and install PopTorch as shown bel
6873
$ pip install -U pip
6974
$ pip install <sdk_path>/poptorch_x.x.x.whl
7075
71-
.. warning:: If, after installing PopTorch, you install a third-party library that requires ``torchvision`` or ``torchaudio`` then that may cause an incompatible version of ``torch`` to be installed.
72-
73-
To prevent this, after installing PopTorch, use the following commands to install compatible versions of ``torchvision`` or ``torchaudio``:
74-
75-
.. code-block:: bash
76-
77-
$ pip install torchvision==0.14.1+cpu --extra-index-url https://download.pytorch.org/whl/cpu
78-
$ pip install torchaudio==0.13.1+cpu --extra-index-url https://download.pytorch.org/whl/cpu
7976
8077
.. _setting_env:
8178

0 commit comments

Comments
 (0)