EuuIia commited on
Commit
1685a74
·
verified ·
1 Parent(s): 0deecfa

Update Dockerfile

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