11 lines
341 B
Plaintext
11 lines
341 B
Plaintext
ARG NVIDIA_PYTORCH_TAG
|
|
FROM nvcr.io/nvidia/pytorch:${NVIDIA_PYTORCH_TAG}
|
|
ARG DEBIAN_FRONTEND=noninteractive
|
|
COPY requirements.txt .
|
|
COPY packages.txt .
|
|
COPY postinstallscript.sh .
|
|
RUN pip install --no-cache-dir -r requirements.txt
|
|
RUN apt-get update && apt-get install -y $(cat packages.txt)
|
|
RUN bash postinstallscript.sh
|
|
WORKDIR /workspace
|