File size: 563 Bytes
704c48c 55639a8 704c48c 55639a8 704c48c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
# Install dependencies
RUN apt-get update && apt-get install -y \
python3 \
python3-pip \
git \
poppler-utils \
&& rm -rf /var/lib/apt/lists/*
# Set the working directory
WORKDIR /app
# Copy the requirements file and install dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
# Copy the app code
COPY app.py .
# Set the environment variable for HF_HUB_DISABLE_SYMLINKS_WARNING
ENV HF_HUB_DISABLE_SYMLINKS_WARNING=1
# Run the app
CMD ["python3", "app.py"]
|