Installation

Install locally

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

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.