|
FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04 |
|
LABEL MAINTAINER="chatGLM" |
|
|
|
COPY . /chatGLM/ |
|
|
|
WORKDIR /chatGLM |
|
|
|
RUN ln -sf /usr/share/zoneinfo/Asia/Shanghai /etc/localtime && echo "Asia/Shanghai" > /etc/timezone |
|
RUN apt-get update -y && apt-get install python3 python3-pip curl libgl1 libglib2.0-0 -y && apt-get clean |
|
RUN curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py && python3 get-pip.py |
|
|
|
RUN pip3 install -r requirements.txt -i https://pypi.mirrors.ustc.edu.cn/simple/ && rm -rf `pip3 cache dir` |
|
|
|
CMD ["python3","-u", "webui.py"] |
|
|