add readme, remove torchvision

2026-04-08 04:08:42 -04:00
parent 263f39e0a3
commit c07187aae1
2 changed files with 62 additions and 1 deletions
@@ -23,7 +23,7 @@ WORKDIR /app

 # Install PyTorch 2.7+ with CUDA 12.8 support (includes Blackwell/sm_120 support)
 RUN python3.11 -m pip install --no-cache-dir \
-    torch torchvision \
+    torch \
    --index-url https://download.pytorch.org/whl/cu128

 # Install auto-gptq pre-built wheel for CUDA 12.8 (avoids compiling from source)