The wheel package of SINGA is installed via pip
. Depending on the hardware environment, here are two ways to install the wheel package.
pip install singa-<version> -f http://singa.apache.org/docs/next/wheel.html
The <version>
field should be replaced, e.g., 3.0.0.dev1
. The available SINGA versions are listed below.
pip install singa-<version>+cuda<cu_version> -f http://singa.apache.org/docs/next/wheel.html
The <version>
field should be replaced with specific SINGA version; The <cu_version>
field should be replaced with specific CUDA version; For example, singa-3.0.0.dev1+cuda10.2
is the full version specification for one package. The available combinations of SINGA version and CUDA version are listed below.
Note: the Python version of your local Python environment will be used to find the corresponding wheel package. For example, if your local Python is 3.6, then the wheel package compiled on Python 3.6 will be selected by pip and installed. In fact, the wheel file's name include SINGA version, CUDA version and Python version. Therefore, pip
knows which wheel file to download and install.
Refer to the comments at the top of the setup.py
file.