Files
heterogeneous-distributed-t…/Dockerfile.linting
tianyutong d6ce507681 Initial Commit of Megatron-LM-0.8.0
Change-Id: Ifb4c061207ee2644a21e161ad52fc6ff40564e39
2025-05-23 09:54:48 +08:00

17 lines
331 B
Docker

# syntax=docker/dockerfile:experimental
ARG FROM_IMAGE_NAME
FROM $FROM_IMAGE_NAME
ENV DEBIAN_FRONTEND=noninteractive
RUN sed -i -e 's/^APT/# APT/' -e 's/^DPkg/# DPkg/' \
/etc/apt/apt.conf.d/docker-clean
RUN pip3 install --no-cache-dir \
black==24.4.2 \
isort
COPY . /opt/megatron-lm
WORKDIR /opt/megatron-lm