mirror of
https://github.com/tencentmusic/cube-studio.git
synced 2024-12-21 06:19:31 +08:00
11 lines
428 B
Docker
11 lines
428 B
Docker
FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime
|
|
|
|
|
|
COPY . /NER
|
|
|
|
RUN python3 -m pip install sklearn_crfsuite -i https://mirrors.aliyun.com/pypi/simple
|
|
RUN python3 -m pip install minio -i https://mirrors.aliyun.com/pypi/simple
|
|
RUN python3 -m pip install tqdm -i https://mirrors.aliyun.com/pypi/simple
|
|
RUN python3 -m pip install transformers -i https://mirrors.aliyun.com/pypi/simple
|
|
|
|
ENTRYPOINT ["python3", "/NER/main4argo.py"] |