Skip to content

Commit 6d8a65c

Browse files
committed
update docker image name
Signed-off-by: ZX-ModelCloud <zx@modelcloud.ai>
1 parent e9e87e6 commit 6d8a65c

1 file changed

Lines changed: 5 additions & 16 deletions

File tree

.github/workflows/unit_tests.yml

Lines changed: 5 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -60,9 +60,9 @@ env:
6060
# must include 8.0 in addition to the newer targets.
6161
TORCH_CUDA_ARCH_LIST: 8.0;8.6;8.9;9.0;12.0
6262
PYTORCH_ALLOC_CONF: 'expandable_segments:True'
63-
RUNNER: 10.0.13.254
63+
RUNNER: 10.0.13.31
6464
XEON5: 10.0.14.249
65-
GPU_ALLOCATOR_URL: http://10.0.13.254/gpu
65+
GPU_ALLOCATOR_URL: http://10.0.13.31/gpu
6666
LOGBAR_ANIMATION: '0'
6767
CUDA_VERSION: 130
6868
UV_TORCH_BACKEND: cu130
@@ -231,7 +231,7 @@ jobs:
231231
check-torch:
232232
runs-on: [ self-hosted, xeon5 ]
233233
container:
234-
image: 10.0.13.254:5000/nvidia/cuda:130-ubuntu24.04_0325
234+
image: 10.0.13.31:5000/nvidia/cuda:130-ubuntu24.04_0612
235235
options: --device /dev/dri --ipc=host --runtime=nvidia --gpus all
236236
volumes:
237237
- /monster/ci/env/entrypoint.sh:/entrypoint.sh
@@ -279,7 +279,7 @@ jobs:
279279
fi
280280
if python -VV 2>&1 | grep -q '3\.14.*free-threading'; then
281281
echo "[INFO] Python 3.14t detected, installing torchao..."
282-
uv pip install http://10.0.13.254/files/torchao-0.18.0+git13cd013d6-cp314-cp314t-linux_x86_64.whl
282+
uv pip install http://10.0.13.31/files/torchao-0.18.0+git13cd013d6-cp314-cp314t-linux_x86_64.whl
283283
else
284284
echo "[INFO] Current Python is not 3.14t, skip torchao."
285285
fi
@@ -310,24 +310,13 @@ jobs:
310310
fi
311311
if python -VV 2>&1 | grep -q '3\.14.*free-threading'; then
312312
echo "[INFO] Python 3.14t detected, installing torchao..."
313-
uv pip install http://10.0.13.254/files/torchao-0.18.0+git13cd013d6-cp314-cp314t-linux_x86_64.whl
313+
uv pip install http://10.0.13.31/files/torchao-0.18.0+git13cd013d6-cp314-cp314t-linux_x86_64.whl
314314
else
315315
echo "[INFO] Current Python is not 3.14t, skip torchao."
316316
fi
317317
318318
${{ matrix.install_cmd }}
319319
320-
debug-docker:
321-
runs-on: [ self-hosted, xeon5 ]
322-
steps:
323-
- run: |
324-
echo "RUNNER_NAME=$RUNNER_NAME"
325-
echo "RUNNER_OS=$RUNNER_OS"
326-
hostname
327-
hostname -I || true
328-
ip -br addr || true
329-
docker info | sed -n '/Insecure Registries:/,/Live Restore Enabled:/p'
330-
331320
setuptools-compatibility:
332321
uses: ./.github/workflows/setuptools_compatibility_reusable.yml
333322
with:

0 commit comments

Comments
 (0)