使用 Nvidia 容器进行多阶段构建

Question

我想通过使用 docker multi-stage 来压平 docker 镜像。我的 Dockerfile 是这样的：

FROM nvcr.io/nvidia/l4t-tensorrt:r8.2.1-runtime as build

# ...

FROM scratch
COPY --from=build / /
CMD ["/bin/bash"]

但是，如果我尝试运行此图像（即使我使用

--runtime nvidia

），Nvidia 库将不会在运行时安装。我该怎么办？

Answer 1

我通过在最新阶段添加

ENV

指令来解决。可以使用

docker inspect nvcr.io/nvidia/l4t-tensorrt:r8.2.1-runtime

检索变量

Answer 2

从 Tech sslassh NVIDIA CUDA 运行时图像生成应用程序开始。 将构建的应用程序从构建阶段复制到运行时阶段。 设置运行应用程序所需的任何必要的环境变量（例如 LD_LIBRARY_PATH）。