23 lines
411 B
Docker
23 lines
411 B
Docker
|
FROM nvidia/cuda:12.2.0-runtime-ubuntu22.04
|
||
|
|
|
||
|
|
RUN apt update && apt install -y \
|
||
|
|
python3.10 \
|
||
|
|
python3-pip \
|
||
|
|
curl \
|
||
|
|
pciutils \
|
||
|
|
lshw \
|
||
|
|
ffmpeg \
|
||
|
|
&& rm -rf /var/lib/apt/lists/*
|
||
|
|
|
||
|
|
WORKDIR /app
|
||
|
|
|
||
|
|
COPY requirements.txt ./
|
||
|
|
RUN pip install --no-cache-dir -r requirements.txt
|
||
|
|
|
||
|
|
COPY . .
|
||
|
|
|
||
|
|
ENV NVIDIA_VISIBLE_DEVICES=all
|
||
|
|
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
|
||
|
|
|
||
|
|
CMD ["python3", "app/worker.py"]
|