mirror of https://github.com/THUDM/CodeGeeX.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
2.8 KiB
Docker
43 lines
2.8 KiB
Docker
FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime
|
|
|
|
RUN cp /etc/apt/sources.list /etc/apt/sources.list.bak \
|
|
&& sed -i "s@http://.*archive.ubuntu.com@https://mirrors.tuna.tsinghua.edu.cn@g" /etc/apt/sources.list \
|
|
&& sed -i "s@http://.*security.ubuntu.com@https://mirrors.tuna.tsinghua.edu.cn@g" /etc/apt/sources.list \
|
|
&& pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple \
|
|
&& apt-get update && apt-get install -y curl npm git nano \
|
|
&& pip install fire zmq transformers tokenizers \
|
|
&& mkdir /workspace/download
|
|
|
|
RUN curl -o /workspace/download/go.tar.gz -SL https://go.dev/dl/go1.18.4.linux-amd64.tar.gz \
|
|
&& tar -zxf /workspace/download/go.tar.gz -C /usr/local && rm /workspace/download/go.tar.gz
|
|
ENV PATH=/bin:/usr/local/go/bin:$PATH
|
|
|
|
RUN curl -o /workspace/download/node.tar.gz -SL https://nodejs.org/download/release/v16.14.0/node-v16.14.0-linux-x64.tar.gz \
|
|
&& mkdir -p /usr/local/lib/nodejs && tar -zxf /workspace/download/node.tar.gz -C /usr/local/lib/nodejs && mv /usr/local/lib/nodejs/node-v16.14.0-linux-x64 /usr/local/lib/nodejs/node \
|
|
&& rm /workspace/download/node.tar.gz && npm install -g js-md5@0.7.3
|
|
ENV PATH=/usr/local/lib/nodejs/node/bin:$PATH
|
|
ENV NODE_PATH=/usr/local/lib/node_modules
|
|
|
|
RUN apt-get install -y build-essential && apt-get install -y g++ \
|
|
&& curl -o /workspace/download/boost_1_71_0.tar.gz -SL https://boostorg.jfrog.io/artifactory/main/release/1.71.0/source/boost_1_71_0.tar.gz \
|
|
&& tar -zxf /workspace/download/boost_1_71_0.tar.gz && rm /workspace/download/boost_1_71_0.tar.gz && cd boost_1_71_0 \
|
|
&& ./bootstrap.sh --prefix=/usr/ && ./b2 && ./b2 install \
|
|
&& cd .. && rm -r boost_1_71_0
|
|
RUN curl -o /workspace/download/openssl.tar.gz -SL https://www.openssl.org/source/old/3.0/openssl-3.0.0.tar.gz \
|
|
&& tar -zxf /workspace/download/openssl.tar.gz && cd openssl-3.0.0 && ./Configure && make && make install \
|
|
&& rm /workspace/download/openssl.tar.gz && rm -r /workspace/openssl-3.0.0
|
|
ENV PATH=/usr/bin/openssl:$PATH
|
|
|
|
RUN curl -o /workspace/download/jdk.tar.gz -SL https://download.oracle.com/java/18/latest/jdk-18_linux-x64_bin.tar.gz \
|
|
&& mkdir /usr/java && tar -zxf /workspace/download/jdk.tar.gz -C /usr/java && rm /workspace/download/jdk.tar.gz \
|
|
&& java_path=`ls /usr/java/${path}` && echo "export JAVA_HOME=/usr/java/${java_path}" >> ~/.profile
|
|
|
|
RUN git clone https://github.com/THUDM/CodeGeeX.git && pip install -e /workspace/CodeGeeX \
|
|
&& tar -zxf /workspace/CodeGeeX/codegeex/benchmark/humaneval-x/go/evaluation/vendor.tar.gz -C /workspace/CodeGeeX/codegeex/benchmark/humaneval-x/go/evaluation
|
|
|
|
SHELL ["sh", "-lc"]
|
|
RUN update-alternatives --install /usr/bin/java java $JAVA_HOME/bin/java 20000 \
|
|
&& update-alternatives --install /usr/bin/javac javac $JAVA_HOME/bin/javac 20000 \
|
|
&& rm -r /workspace/download \
|
|
&& env
|