Installation

Install locally

To install Bagua, besides your deep learning framework (like PyTorch), you need the following dependencies installed on your system:

• CUDA Toolkit, with CUDA version >= 10.1
• Rust Compiler
• MPI >= 3.0, for example Open MPI
• hwloc >= 2.0
• CMake >= 3.17

We provide an automatic installation script for Ubuntu. Just run the following command to install Bagua and above libraries (except for CUDA, you should always install CUDA by yourself):

curl -Ls https://raw.githubusercontent.com/BaguaSys/bagua/master/install.sh | sudo bash

If you already have dependencies installed on your system, you can install the bagua python package only:

# release version
python3 -m pip install bagua

# develop version (git master)
python3 -m pip install git+https://github.com/BaguaSys/bagua.git

Use Docker image

We provide Docker image with Bagua installed based on official PyTorch images. You can find them on DockerHub.