Update Dockerfile
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
@@ -18,8 +18,8 @@ ENV HF_HOME=/app/.cache \
|
|
18 |
# ✅ Pip güncelle
|
19 |
RUN pip install --upgrade pip
|
20 |
|
21 |
-
# ✅ Uyumlu Torch sürümü yükle (A100 + CUDA 11.8 için)
|
22 |
-
RUN pip install torch==2.1.2 torchvision --index-url https://download.pytorch.org/whl/cu118
|
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 + Triton sürümü yükle (A100 + CUDA 11.8 için)
|
22 |
+
RUN pip install torch==2.1.2 torchvision triton --index-url https://download.pytorch.org/whl/cu118
|
23 |
|
24 |
# ✅ Gereksinim dosyalarını yükle
|
25 |
COPY requirements.txt .
|