INSTALLATION.md

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

bazel build //:spu_wheel -c opt
pip install bazel-bin/spu-*.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

bazel build //:spu_wheel -c opt --config=gpu

Build with specified python version

# build with python 3.10
bazel build //:spu_wheel -c opt --@rules_python//python/config_settings:python_version=3.10

# build with python 3.11
bazel build //:spu_wheel -c opt --@rules_python//python/config_settings:python_version=3.11