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
