17 lines
513 B
Docker
17 lines
513 B
Docker
FROM pytorch/pytorch:2.9.1-cuda12.8-cudnn9-runtime
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive \
|
|
PYTHONUNBUFFERED=1 \
|
|
TOKENIZERS_PARALLELISM=false \
|
|
HF_HOME=/workspace/.cache/huggingface \
|
|
PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
|
|
|
|
RUN apt-get update \
|
|
&& apt-get install -y --no-install-recommends git ca-certificates \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
COPY requirements.txt /tmp/requirements.txt
|
|
RUN python -m pip install --no-cache-dir -r /tmp/requirements.txt
|
|
|
|
WORKDIR /workspace/recipe
|