3. PyTorch setup¶
You should activate the Python virtual environment you created for PyTorch (Section 2.6, Working with a Python virtual environment) before performing the setup in this section.
This section describes how to install a wheel file containing PopTorch, a set of extensions to enable PyTorch models to run on the IPU.
Install the PopTorch wheel file using a command similar to the following:
$ cd $SDKDIR $ python -m pip install poptorch-X+X_X.whl
Section 2.1, Define SDK location describes how to define the
SDKDIR environment variable.
An example of the filename for the PopTorch wheel file is:
To confirm that PopTorch has been installed, you can use
pip list, which should include the
poptorch package in the output.
Package Version ------------- ---------- numpy 1.19.5 packaging 21.3 Pillow 8.4.0 pip 21.3.1 pkg_resources 0.0.0 pluggy 0.13.1 poptorch 2.3.0+30608 wheel 0.37.0
You can also test that the module has been installed correctly by attempting to import it in Python, for example:
$ python3 -c "import poptorch; print(poptorch.__version__)"