docs | |
---|---|
tests | |
package |
A Distributed Deep Learning package for PyTorch.
- Free software: BSD 2-Clause License
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.
https://distdl.readthedocs.io/
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.