wehub-resource-sync
ec436095dd
Book-CI / test (macos-latest) (push) Has been cancelled
Book-CI / test (ubuntu-latest) (push) Has been cancelled
Book-CI / test (windows-latest) (push) Has been cancelled
Release Fake Tag / publish (push) Has been cancelled
Deploy / deploy (macos-latest) (push) Has been cancelled
Deploy / deploy (ubuntu-latest) (push) Has been cancelled
Deploy / deploy (windows-latest) (push) Has been cancelled
Release to PyPI / Build & publish sglang-kt (push) Has been cancelled
Release to PyPI / Build kt-kernel (Python 3.11) (push) Has been cancelled
Release to PyPI / Build kt-kernel (Python 3.12) (push) Has been cancelled
Release to PyPI / Publish kt-kernel to PyPI (push) Has been cancelled
20 lines
499 B
Docker
20 lines
499 B
Docker
FROM pytorch/pytorch:2.5.1-cuda12.1-cudnn9-devel as compile_server
|
|
WORKDIR /workspace
|
|
ENV CUDA_HOME /usr/local/cuda
|
|
RUN <<EOF
|
|
apt update -y && apt install -y --no-install-recommends \
|
|
git \
|
|
wget \
|
|
vim \
|
|
gcc \
|
|
g++ \
|
|
cmake &&
|
|
rm -rf /var/lib/apt/lists/* &&
|
|
pip install --upgrade pip &&
|
|
pip install ninja pyproject numpy cpufeature &&
|
|
pip install flash-attn &&
|
|
cp /usr/lib/x86_64-linux-gnu/libstdc++.so.6 /opt/conda/lib/
|
|
EOF
|
|
# Set the default shell to bash
|
|
CMD ["/bin/bash"]
|