File size: 1,758 Bytes
51be264 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
FROM nvidia/cuda:12.5.1-devel-ubuntu22.04
ARG PYTHON_VERSION=3.11
ARG http_proxy
ARG https_proxy
RUN apt-get update
RUN apt-get install -y \
locales \
build-essential \
git \
git-lfs \
vim \
cmake \
pkg-config \
zlib1g-dev libncurses5-dev \
libgdbm-dev libnss3-dev libssl-dev libreadline-dev libffi-dev wget \
liblzma-dev libsqlite3-dev libbz2-dev
RUN apt-get clean
ENV LANG=en_US.UTF-8
ENV LANGUAGE=en_US:en
ENV LC_ALL=en_US.UTF-8
RUN sed -i '/en_US.UTF-8/s/^# //g' /etc/locale.gen && locale-gen
ENV PYENV_ROOT=/root/.pyenv
ENV PATH="$PYENV_ROOT/bin/:$PATH"
RUN /usr/bin/echo -e '#!/bin/bash\neval "$(pyenv init -)"\neval "$(pyenv virtualenv-init -)"\ncd /moe_peft\nbash' | tee /opt/init.sh \
&& chmod +x /opt/init.sh \
&& /usr/bin/echo -e 'export PYENV_ROOT=/root/.pyenv' >> ~/.bashrc \
&& /usr/bin/echo -e 'export PATH=/root/.pyenv/bin:$PATH' >> ~/.bashrc \
&& /usr/bin/echo -e 'eval "$(pyenv init -)"' >> ~/.bashrc \
&& /usr/bin/echo -e 'eval "$(pyenv virtualenv-init -)"' >> ~/.bashrc \
&& git clone https://github.com/pyenv/pyenv.git /root/.pyenv \
&& git clone https://github.com/pyenv/pyenv-virtualenv.git /root/.pyenv/plugins/pyenv-virtualenv \
&& cd /root/.pyenv && src/configure && make -C src \
&& eval "$(pyenv init -)" \
&& eval "$(pyenv virtualenv-init -)"
RUN . ~/.bashrc \
&& pyenv install $PYTHON_VERSION \
&& pyenv global $PYTHON_VERSION \
&& git clone https://github.com/TUDB-Labs/MoE-PEFT /moe_peft \
&& cd /moe_peft \
&& pyenv virtualenv $PYTHON_VERSION moe_peft \
&& pyenv local moe_peft \
&& pip install -r ./requirements.txt --upgrade --no-compile --no-cache-dir
WORKDIR /moe_peft
CMD ["/bin/bash", "/opt/init.sh"] |