FROM rocm/pytorch:rocm5.0.1_ubuntu18.04_py3.7_pytorch_1.10.0

# install latest released version of deepspeed
RUN pip install deepspeed && \
    ds_report