[Sincronização Automática] - 2025-09-23 16:17:28

Dockerfile

@@ -64,9 +64,19 @@ RUN pip install \
     torch>=2.8.0+cu128 \
     torchvision \
     torchaudio \
-    --index-url https://download.pytorch.org/whl/cu128 \
-    packaging wheel ninja setuptools
-
+    --index-url https://download.pytorch.org/whl/cu128 
+    
+RUN pip install  \
+    packaging \
+    ninja \
+    cmake \
+    pybind11 \
+    scikit-build \
+    cython \
+    numpy
+    
+    
+    
 # 2. Instala a wheel pré-compilada do Flash Attention 2.8.0
 #    Esta é a forma mais rápida e estável de instalar, evitando compilação demorada.
 #    Esta wheel é compatível com PyTorch 2.8.0 e CUDA 12.x