13 lines
336 B
Docker
13 lines
336 B
Docker
FROM corex:4.3.8
|
|
|
|
WORKDIR /root
|
|
|
|
ADD . /root/
|
|
|
|
COPY requirements.txt /root
|
|
RUN pip install -r requirements.txt -i https://nexus.4pd.io/repository/pypi-all/simple
|
|
# 安装torch是为了提供cuda库环境
|
|
RUN pip install transformers==4.51.3 -i https://nexus.4pd.io/repository/pypi-all/simple
|
|
|
|
ENTRYPOINT ["python3"]
|
|
CMD ["./main_qa.py"] |