carlex3321 commited on
Commit
dcb90f3
·
verified ·
1 Parent(s): 7fe4e97

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -5
Dockerfile CHANGED
@@ -67,7 +67,7 @@ RUN mkdir -p /home/user/.cache/models && ln -sf /home/user/.cache/models /app/mo
67
 
68
  # ---------------- Sistema & Python ----------------
69
  RUN apt-get update && apt-get install -y --no-install-recommends \
70
- build-essential cmake git git-lfs curl wget ffmpeg ninja-build \
71
  python3.10 python3.10-dev python3.10-distutils python3-pip \
72
  && apt-get clean && rm -rf /var/lib/apt/lists/*
73
 
@@ -75,7 +75,10 @@ RUN ln -sf /usr/bin/python3.10 /usr/bin/python3 && \
75
  ln -sf /usr/bin/python3.10 /usr/bin/python && \
76
  python3 -m pip install --upgrade pip
77
 
78
-
 
 
 
79
 
80
  # FlashAttention 2.8.x
81
  RUN pip install flash-attn==2.8.3 --no-build-isolation || \
@@ -113,9 +116,6 @@ RUN pip install --index-url https://download.pytorch.org/whl/cu128 \
113
  torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0+cu128
114
 
115
 
116
- RUN apt-get update && apt-get install -y \
117
- python3-pybind11 ninja-build cmake gcc g++ python3-dev \
118
- && pip install --upgrade pip setuptools wheel pybind11 ninja cmake
119
 
120
  RUN pip uninstall -y triton || true && \
121
  pip install -v --no-build-isolation triton==3.4.0
 
67
 
68
  # ---------------- Sistema & Python ----------------
69
  RUN apt-get update && apt-get install -y --no-install-recommends \
70
+ build-essential cmake git git-lfs curl wget ffmpeg packaging ninja-build \
71
  python3.10 python3.10-dev python3.10-distutils python3-pip \
72
  && apt-get clean && rm -rf /var/lib/apt/lists/*
73
 
 
75
  ln -sf /usr/bin/python3.10 /usr/bin/python && \
76
  python3 -m pip install --upgrade pip
77
 
78
+ RUN apt-get update && apt-get install -y \
79
+ python3-pybind11 ninja-build cmake gcc g++ python3-dev \
80
+ && pip install --upgrade pip setuptools wheel pybind11 ninja cmake
81
+
82
 
83
  # FlashAttention 2.8.x
84
  RUN pip install flash-attn==2.8.3 --no-build-isolation || \
 
116
  torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0+cu128
117
 
118
 
 
 
 
119
 
120
  RUN pip uninstall -y triton || true && \
121
  pip install -v --no-build-isolation triton==3.4.0