FROM nvidia/cuda:11.7.0-base-ubuntu18.04
#拷贝业务程序
COPY --chown=1000:1000 app /app
ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/app/lib
WORKDIR /app
ENTRYPOINT ["./train"]