forked from iflytek/aiges
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathDockerfile_gpu
31 lines (18 loc) · 891 Bytes
/
Dockerfile_gpu
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
FROM artifacts.iflytek.com/docker-private/atp/aiges-gpu:10.1-3.9.13-ubuntu1804-v1.0.0
RUN pip3 config set global.index-url https://pypi.mirrors.ustc.edu.cn/simple/
RUN pip3 install torch==1.10 torchvision
RUN apt install -y libgl1-mesa-glx && pip3 install openmim && \
mim install mmcv-full && \
pip install mmdet==2.24.0
copy mmocr /home/mmocr
RUN cd /home/mmocr/ && pip3 install -e .
RUN pip3 install iflags
COPY wrapper.py /home/mmocr
# 拷贝mmocr配置文件到 /home/aiges, 因为工作目录位于 /home/aiges
RUN cp -ra /home/mmocr/configs /home/aiges
ENV PYTHONPATH=$PYTHONPATH:/home/mmocr
ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/wrapper/wrapper_lib
COPY xtest.toml /home/aiges
COPY xtest /home/aiges/xtest
COPY aiges.toml /home/aiges
CMD ["sh", "-c", "./AIservice -m=0 -c=aiges.toml -s=svcName -u=http://companion.xfyun.iflytek:6868 -p=AIaaS -g=dx"]