Dockerfile.rocm 118 B

12345
  1. FROM rocm/pytorch:latest
  2. # install latest released version of deepspeed
  3. RUN pip install deepspeed && \
  4. ds_report