mirror of
https://github.com/kvcache-ai/custom_flashinfer.git
synced 2026-07-17 09:08:15 +00:00
aa9c5d930f
This enables users to develop inside a prebuilt container. This can save a lot of efforts of setting up cuda toolkit etc. ref: https://code.visualstudio.com/docs/devcontainers/containers
59 lines
1.8 KiB
Docker
59 lines
1.8 KiB
Docker
FROM nvidia/cuda:12.4.0-devel-ubuntu22.04
|
|
|
|
# Update package lists and install system dependencies
|
|
RUN apt-get update && apt-get install -y \
|
|
curl \
|
|
git \
|
|
clang-format \
|
|
libibverbs-dev \
|
|
librdmacm-dev \
|
|
rdma-core \
|
|
libnuma-dev \
|
|
vim \
|
|
openmpi-bin \
|
|
libopenmpi-dev \
|
|
zsh \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Install oh-my-zsh
|
|
RUN sh -c "$(curl -fsSL https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/tools/install.sh)" "" --unattended
|
|
|
|
# Install powerlevel10k theme
|
|
RUN git clone --depth=1 https://github.com/romkatv/powerlevel10k.git ${ZSH_CUSTOM:-$HOME/.oh-my-zsh/custom}/themes/powerlevel10k
|
|
|
|
# Install zsh-autosuggestions
|
|
RUN git clone https://github.com/zsh-users/zsh-autosuggestions ${ZSH_CUSTOM:-~/.oh-my-zsh/custom}/plugins/zsh-autosuggestions
|
|
|
|
# Configure zsh
|
|
RUN sed -i 's/ZSH_THEME="robbyrussell"/ZSH_THEME="powerlevel10k\/powerlevel10k"/' ~/.zshrc && \
|
|
sed -i 's/plugins=(git)/plugins=(git zsh-autosuggestions)/' ~/.zshrc
|
|
|
|
# Create a non-root user
|
|
ARG USERNAME=devuser
|
|
ARG USER_UID=1003
|
|
ARG USER_GID=$USER_UID
|
|
|
|
# Create the user
|
|
RUN groupadd --gid $USER_GID $USERNAME \
|
|
&& useradd --uid $USER_UID --gid $USER_GID -m $USERNAME \
|
|
# [Optional] Add sudo support
|
|
&& apt-get update \
|
|
&& apt-get install -y sudo \
|
|
&& echo $USERNAME ALL=\(root\) NOPASSWD:ALL > /etc/sudoers.d/$USERNAME \
|
|
&& chmod 0440 /etc/sudoers.d/$USERNAME \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Copy zsh configuration to the new user's home
|
|
RUN cp -r /root/.oh-my-zsh /home/$USERNAME/.oh-my-zsh && \
|
|
cp /root/.zshrc /home/$USERNAME/.zshrc && \
|
|
chown -R $USERNAME:$USERNAME /home/$USERNAME/.oh-my-zsh && \
|
|
chown $USERNAME:$USERNAME /home/$USERNAME/.zshrc
|
|
|
|
# Switch to non-root user
|
|
USER $USERNAME
|
|
WORKDIR /home/$USERNAME
|
|
|
|
# Set zsh as default shell
|
|
ENV SHELL=/bin/zsh
|
|
CMD [ "zsh" ]
|