2025-02-23 16:44:08 +01:00
|
|
|
FROM nvidia/cuda:12.2.0-runtime-ubuntu22.04
|
|
|
|
|
|
2025-02-23 18:14:46 +01:00
|
|
|
RUN apt update && apt install -y python3.10 python3-pip curl pciutils lshw ffmpeg && \
|
2025-02-23 16:44:08 +01:00
|
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
|
|
|
|
|
|
WORKDIR /app
|
2025-02-23 17:35:00 +01:00
|
|
|
|
2025-02-23 16:44:08 +01:00
|
|
|
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", "main.py"]
|
