ciyidogan commited on
Commit
ebf9b74
·
verified ·
1 Parent(s): 76b3189

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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 + 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 .
 
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 .