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"]