From 5bc7a165a9a7694e75dc6a165382a714273bfe1b Mon Sep 17 00:00:00 2001
From: Untone <anton.rewin@gmail.com>
Date: Fri, 27 Sep 2024 00:42:03 +0300
Subject: [PATCH] image12

---
 Dockerfile | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 1caa9c5..5812ceb 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,17 +1,13 @@
 # Stage 1: Build stage
-FROM python:slim AS builder
+FROM pytorch:slim AS builder
 
 WORKDIR /app
 
 COPY requirements.txt .
 
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    gcc \
-    libffi-dev \
-    libssl-dev
-RUN pip install triton
-RUN pip install --no-cache-dir -r requirements.txt \
-    && rm -rf /var/lib/apt/lists/*
+RUN apt-get update && apt-get install -y --no-install-recommends gcc libffi-dev libssl-dev
+
+RUN pip install --no-cache-dir -r requirements.txt && rm -rf /var/lib/apt/lists/*
 
 # Stage 2: Final stage
 FROM python:slim