Skip to content

distdl/distdl

Repository files navigation

Overview

docs Documentation Status
tests
package

A Distributed Deep Learning package for PyTorch.

  • Free software: BSD 2-Clause License

Installation

You can install the in-development version with:

pip install https://github.com/distdl/distdl/archive/master.zip

DistDL will be available on pip shortly.

Documentation

https://distdl.readthedocs.io/

Development

To run the all tests run:

mpirun -np 20 python -m mpi4py -m pytest --with-mpi -rsa tests

Substitute mpiexec or srun as correct for your system.