Dockerfile.gpu 249 字节
Newer Older
G
gangliao 已提交
1 2 3 4 5 6 7 8
FROM paddlepaddle/paddle:latest-gpu

ENV PARAMETER_TAR_PATH=/data/param.tar \
    TOPOLOGY_FILE_PATH=/data/inference_topology.pkl
ADD requirements.txt /root
ADD main.py /root
RUN pip install -r /root/requirements.txt
CMD ["python", "/root/main.py"]