FROM rocm/pytorch:latest
# install latest released version of deepspeed RUN pip install deepspeed && \ ds_report