3. PyTorch setup
This section describes how to install a wheel file containing PopTorch, a set of extensions to enable PyTorch models to run on the IPU.
You should activate the Python virtual environment you created for PyTorch (Section 2.7, Create a Python virtual environment) before performing the setup in this section.
Install the PopTorch wheel file using a command similar to the following:
$ python3 -m pip install $SDK_DIR/poptorch-X+X_X.whl
$SDK_DIR is the location of the Poplar SDK (Section 2.4, Define environment variables).
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 3.0.0+86945 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__)"