id: wheel title: Wheel

Installation Instructions

The wheel package of SINGA is installed via pip. Depending on the hardware environment, here are two ways to install the wheel package.

  • CPU only
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.

  • With CUDA and cuDNN
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.

Building Instructions

Refer to the comments at the top of the setup.py file.

SINGA-3.0.0.dev1