image12
This commit is contained in:
12
Dockerfile
12
Dockerfile
@@ -1,17 +1,13 @@
|
|||||||
# Stage 1: Build stage
|
# Stage 1: Build stage
|
||||||
FROM python:slim AS builder
|
FROM pytorch:slim AS builder
|
||||||
|
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
|
|
||||||
COPY requirements.txt .
|
COPY requirements.txt .
|
||||||
|
|
||||||
RUN apt-get update && apt-get install -y --no-install-recommends \
|
RUN apt-get update && apt-get install -y --no-install-recommends gcc libffi-dev libssl-dev
|
||||||
gcc \
|
|
||||||
libffi-dev \
|
RUN pip install --no-cache-dir -r requirements.txt && rm -rf /var/lib/apt/lists/*
|
||||||
libssl-dev
|
|
||||||
RUN pip install triton
|
|
||||||
RUN pip install --no-cache-dir -r requirements.txt \
|
|
||||||
&& rm -rf /var/lib/apt/lists/*
|
|
||||||
|
|
||||||
# Stage 2: Final stage
|
# Stage 2: Final stage
|
||||||
FROM python:slim
|
FROM python:slim
|
||||||
|
Reference in New Issue
Block a user