update git ignore add docker dev container

This commit is contained in:
liam 2025-02-25 17:19:19 +08:00
parent 5474be5299
commit 0ca0b99fab
4 changed files with 57 additions and 8 deletions

19
.devcontainer/Dockerfile Normal file
View file

@ -0,0 +1,19 @@
FROM pytorch/pytorch:2.3.1-cuda12.1-cudnn8-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/* &&
cd ktransformers &&
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"]