Update Dockerfile
Browse files- Dockerfile +5 -1
Dockerfile
CHANGED
|
@@ -56,7 +56,7 @@ RUN ln -sf /data/models /app/models
|
|
| 56 |
|
| 57 |
# ---------------- System & Python ----------------
|
| 58 |
RUN apt-get update && apt-get install -y --no-install-recommends \
|
| 59 |
-
build-essential
|
| 60 |
python3.10 python3.10-dev python3.10-distutils python3-pip \
|
| 61 |
ca-certificates libglib2.0-0 libgl1 \
|
| 62 |
&& apt-get clean && rm -rf /var/lib/apt/lists/*
|
|
@@ -65,6 +65,10 @@ RUN ln -sf /usr/bin/python3.10 /usr/bin/python3 && \
|
|
| 65 |
ln -sf /usr/bin/python3.10 /usr/bin/python && \
|
| 66 |
python3 -m pip install --upgrade pip
|
| 67 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 68 |
# ---------------- PyTorch cu128 (pinned) ----------------
|
| 69 |
RUN pip install --index-url https://download.pytorch.org/whl/cu128 \
|
| 70 |
torch>=2.8.0+cu128 torchvision>=0.23.0+cu128 torchaudio>=2.8.0+cu128
|
|
|
|
| 56 |
|
| 57 |
# ---------------- System & Python ----------------
|
| 58 |
RUN apt-get update && apt-get install -y --no-install-recommends \
|
| 59 |
+
build-essential gosu tree cmake git git-lfs curl wget ffmpeg ninja-build \
|
| 60 |
python3.10 python3.10-dev python3.10-distutils python3-pip \
|
| 61 |
ca-certificates libglib2.0-0 libgl1 \
|
| 62 |
&& apt-get clean && rm -rf /var/lib/apt/lists/*
|
|
|
|
| 65 |
ln -sf /usr/bin/python3.10 /usr/bin/python && \
|
| 66 |
python3 -m pip install --upgrade pip
|
| 67 |
|
| 68 |
+
|
| 69 |
+
RUN curl -L https://github.com/bazelbuild/bazelisk/releases/latest/download/bazelisk-linux-amd64 -o bazel && \
|
| 70 |
+
chmod +x bazel
|
| 71 |
+
|
| 72 |
# ---------------- PyTorch cu128 (pinned) ----------------
|
| 73 |
RUN pip install --index-url https://download.pytorch.org/whl/cu128 \
|
| 74 |
torch>=2.8.0+cu128 torchvision>=0.23.0+cu128 torchaudio>=2.8.0+cu128
|