deepspeedai--deepspeed
wehub-resource-sync
6 行
118 B
Docker
6 行
118 B
Docker
FROM rocm/pytorch:latest
|
|
|
|
# install latest released version of deepspeed
|
|
RUN pip install deepspeed && \
|
|
ds_report
|