FROM nvidia/cuda:11.8.0-devel-ubuntu22.04

RUN apt-get update && apt-get install -y \
	git \
	python3 \
	pip

	
WORKDIR /app

COPY . /app


# upgrade pip
RUN pip3 install --upgrade pip

RUN pip install --no-cache-dir -r requirements.txt

RUN python3 -m spacy download zh_core_web_sm

EXPOSE 7860

CMD ["python3", "pilot/server/webserver.py"]
