dockfile_classify

FROM   nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04

COPY ./   /root/doc_clasification

RUN  apt-get install python3.8

WORKDIR /root/doc_clasification
RUN apt-get install git
RUN git lfs install
RUN git clone https://huggingface.co/GanymedeNil/text2vec-large-chinese
RUN pip3.10 install  -i requirements.txt
RUN pip3.10 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

EXPOSE  8000

CMD   python3.8   file_classify/service.py