Quick Start

First, find the veScale repo.

Installation

bash [repo]/patches/build_pytorch_w_patch.sh

This will compile and install a patched version of PyTorch.

bash [repo]/patches/build_torchdistX_w_patch.sh

This will compile and install a patched version of TorchdistX (based on its master).

pushd python && pip3 install -r requirements.txt && pip3 install -e . && popd

This will install veScale and its dependencies.

Make sure it is in the veScale directory.

docker build .

It may take a while to build the image.

Once the building process is finished, you can docker run with the id.