|
|
|
|
|
FROM nvidia/cuda:11.3.1-runtime-ubuntu20.04 |
|
ARG useProxyNetwork='' |
|
RUN apt-get update |
|
RUN apt-get install -y curl proxychains curl gcc |
|
RUN apt-get install -y git python python3 python-dev python3-dev --fix-missing |
|
|
|
|
|
|
|
RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3.8 |
|
|
|
RUN python3 -m pip install torch --extra-index-url https://download.pytorch.org/whl/cu113 |
|
|
|
WORKDIR /gpt |
|
RUN git clone https://github.com/binary-husky/chatgpt_academic.git |
|
WORKDIR /gpt/chatgpt_academic |
|
RUN git clone https://github.com/OpenLMLab/MOSS.git request_llm/moss |
|
RUN python3 -m pip install -r requirements.txt |
|
RUN python3 -m pip install -r request_llm/requirements_moss.txt |
|
RUN python3 -m pip install -r request_llm/requirements_chatglm.txt |
|
RUN python3 -m pip install -r request_llm/requirements_newbing.txt |
|
|
|
|
|
|
|
|
|
RUN python3 -c 'from check_proxy import warm_up_modules; warm_up_modules()' |
|
|
|
|
|
CMD ["python3", "-u", "main.py"] |
|
|
