wangxiaoman/python-pytorch-huggingface repository overview

Dockfile内容：

from python:3.10

安装pytorch

RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu⁠

transformer安装

RUN pip3 install transformers datasets evaluate accelerate