Dockerfile-cuda 462 B

12345678910111213
  1. FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04
  2. LABEL MAINTAINER="chatGLM"
  3. COPY . /chatGLM/
  4. WORKDIR /chatGLM
  5. RUN apt-get update -y && apt-get install python3 python3-pip curl -y && apt-get clean
  6. RUN curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py && python3 get-pip.py
  7. RUN pip3 install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn && rm -rf `pip3 cache dir`
  8. CMD ["python3","-u", "webui.py"]