Installation Guidelines

Environment

Linux

SPU has been tested with the following settings:

• Ubuntu 22.04
• python3.10
• 8c16g

MacOS

We have conducted some successful preliminary testings on macOS Monterey 14.1 with Apple Silicon.

Docker Image

Please check official Docker image.

Binaries

From PyPI

You could install SPU via the official PyPI package.

pip install spu

From Source

• Install dependencies listed here.
• For CentOS 7 or Ubuntu 22.04, use corresponding dockerfile below as a reference:
  • Ubuntu 22.04 reference
  • CentOS 7 reference
• At the root of repo, run

python setup.py bdist_wheel
pip install dist/*.whl --force-reinstall

• Once GCC/bazel/python/Xcode version or other environment settings have changed, please run the following command to ensure a clean build

bazel clean --expunge

Build with GPU support

export ENABLE_GPU_BUILD=1 && python setup.py bdist_wheel