diff --git a/Dockerfile b/Dockerfile index 5cf47fa..af81ad4 100644 --- a/Dockerfile +++ b/Dockerfile @@ -3,16 +3,25 @@ FROM python:3.11-slim RUN --mount=type=cache,target=/root/.cache/pip pip install -U pip ARG TARGETPLATFORM -RUN apt-get update && apt-get install --no-install-recommends -y curl ffmpeg -RUN if [ "$TARGETPLATFORM" != "linux/amd64" ]; then apt-get install --no-install-recommends -y build-essential ; fi -RUN if [ "$TARGETPLATFORM" != "linux/amd64" ]; then curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y ; fi -ENV PATH="/root/.cargo/bin:${PATH}" -# for deepspeed support - doesn't seem worth it, image +7.5GB, over the 10GB ghcr.io limit, and no noticable gain in speed or VRAM usage? -#RUN curl -O https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/cuda-keyring_1.1-1_all.deb -#RUN dpkg -i cuda-keyring_1.1-1_all.deb && rm cuda-keyring_1.1-1_all.deb -#RUN apt-get update && apt-get install --no-install-recommends -y libaio-dev build-essential cuda-toolkit +RUN <