下载模型文件 https://huggingface.co/facebook/nllb-200-3.3B
sudo docker run -d --name jianbing_nllb --log-opt max-size=1m --gpus=all -p 8080:5000 -v /nllb-200-3.3B:/app/nllb-200-3.3B jianbing/nllb-200
FROM python:3.10-slim
RUN pip3 install torch torchvision torchaudio
RUN python -m pip install --upgrade pip
RUN pip install transformers sentencepiece flask
COPY api.py /api.py
WORKDIR /
EXPOSE 5000
CMD [ "python","api.py" ]
Content type
Image
Digest
sha256:f3a286174…
Size
4.5 GB
Last updated
about 3 years ago
docker pull jianbing/nllb-200