wehub-resource-sync
59a0a3844c
PR Test AMD / cancel-on-close (push) Has been skipped
PR Test NVIDIA ARM / scan (push) Has been skipped
PR Test NVIDIA / cancel-on-close (push) Has been skipped
PR Test AMD / scan (push) Has been skipped
PR Test NVIDIA ARM / cancel-on-close (push) Has been skipped
PR Test NVIDIA / scan (push) Has been skipped
Release Docker Images / build (cu129-torch-2.11.0) (push) Has been skipped
Release Docker Images / build (cu130-torch-2.11.0) (push) Has been skipped
Release PyPI / publish (push) Has been skipped
Scheduler Python Test / test (push) Successful in 27m19s
Docs / build (push) Successful in 28m8s
Scheduler C++ Test / test (push) Successful in 28m19s
Scheduler C++ Test / test-flat (push) Successful in 28m18s
Docs / deploy (push) Has been cancelled
PR Test AMD / finish (push) Has been cancelled
PR Test NVIDIA / finish (push) Has been cancelled
PR Test NVIDIA ARM / finish (push) Has been cancelled
PR Test NVIDIA ARM / ${{ matrix.name }} (${{ matrix.runner }}) (push) Has been cancelled
PR Test AMD / ${{ matrix.name }} (${{ matrix.runner }}) (push) Has been cancelled
PR Test NVIDIA / ${{ matrix.name }} (${{ matrix.runner }}) (push) Has been cancelled
25 lines
800 B
Docker
25 lines
800 B
Docker
ARG RUNNER_TAG=cu130-torch-2.11.0
|
|
FROM lightseekorg/tokenspeed-runner:${RUNNER_TAG}
|
|
|
|
USER root
|
|
|
|
ARG MAX_JOBS=16
|
|
ENV PIP_BREAK_SYSTEM_PACKAGES=1
|
|
|
|
WORKDIR /workspace
|
|
|
|
COPY . /workspace
|
|
|
|
# libnuma1: runtime dep of torch_memory_saver (the Sleep/Wake Up memory saver),
|
|
# which is a hard dependency of the python package; the base runner image lacks it.
|
|
RUN apt-get update && apt-get install -y --no-install-recommends libssl-dev libopenmpi-dev libnuma1 && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
ARG CUDA_ARCH_LIST="9.0a 10.0a"
|
|
RUN MAX_JOBS=${MAX_JOBS} FLASHINFER_CUDA_ARCH_LIST="${CUDA_ARCH_LIST}" TOKENSPEED_KERNEL_BACKEND=cuda \
|
|
pip install tokenspeed-kernel/python/ --no-build-isolation && \
|
|
pip install tokenspeed-scheduler/ && \
|
|
pip install "./python" --no-build-isolation
|
|
|
|
CMD ["/bin/bash"]
|