Update Dockerfile
Browse files- Dockerfile +3 -2
Dockerfile
CHANGED
@@ -18,8 +18,9 @@ ENV HF_HOME=/app/.cache \
|
|
18 |
# ✅ Pip güncelle
|
19 |
RUN pip install --upgrade pip
|
20 |
|
21 |
-
# ✅ Torch
|
22 |
-
RUN pip install torch==2.
|
|
|
23 |
|
24 |
# ✅ Gereksinim dosyalarını yükle
|
25 |
COPY requirements.txt .
|
|
|
18 |
# ✅ Pip güncelle
|
19 |
RUN pip install --upgrade pip
|
20 |
|
21 |
+
# ✅ Uyumlu Torch sürümü + pip triton yükle
|
22 |
+
RUN pip install torch==2.1.2 torchvision --index-url https://download.pytorch.org/whl/cu118
|
23 |
+
RUN pip install triton
|
24 |
|
25 |
# ✅ Gereksinim dosyalarını yükle
|
26 |
COPY requirements.txt .
|