9 lines
228 B
Docker
9 lines
228 B
Docker
FROM python:3.11-slim
|
|
WORKDIR /app
|
|
COPY . .
|
|
# Argument for specifying the GPU type, defaulting to "other"
|
|
ARG gpu_type=other
|
|
RUN pip install --no-cache-dir -r requirements_${gpu_type}.txt
|
|
EXPOSE 7851
|
|
CMD ["python", "script.py"]
|