ciyidogan commited on
Commit
97faa36
·
verified ·
1 Parent(s): b908300

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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 2.0.1 + cu118 yükle (Unsloth ile uyumlu)
22
- RUN pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 --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 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 .