mirror of
https://github.com/tencentmusic/cube-studio.git
synced 2024-12-27 06:29:10 +08:00
11 lines
443 B
Docker
11 lines
443 B
Docker
FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime
|
|
|
|
|
|
COPY . /NER
|
|
|
|
RUN python3 -m pip install sklearn_crfsuite -i https://pypi.tuna.tsinghua.edu.cn/simple
|
|
RUN python3 -m pip install minio -i https://pypi.tuna.tsinghua.edu.cn/simple
|
|
RUN python3 -m pip install tqdm -i https://pypi.tuna.tsinghua.edu.cn/simple
|
|
RUN python3 -m pip install transformers -i https://pypi.tuna.tsinghua.edu.cn/simple
|
|
WORKDIR /NER
|
|
ENTRYPOINT ["python3", "main4argo.py"] |