16 lines
363 B
Docker
16 lines
363 B
Docker
|
FROM nvidia/cuda:12.2.0-runtime-ubuntu22.04
|
||
|
|
|
||
|
|
RUN apt update && apt install -y python3.9 python3-pip curl pciutils lshw && \
|
||
|
|
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", "main.py"]
|