add Dockerfiles
This commit is contained in:
parent
65c78a31d3
commit
06d6badf28
|
@ -0,0 +1,15 @@
|
|||
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"]
|
|
@ -0,0 +1,10 @@
|
|||
FROM python:3.9-slim
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
COPY requirements.txt ./
|
||||
RUN pip install --no-cache-dir -r requirements.txt
|
||||
|
||||
COPY . .
|
||||
|
||||
CMD ["python", "main.py"]
|
Loading…
Reference in New Issue