如何解决Nvidia Jetson上CUDA C++应用Docker镜像构建失败问题？

Docker镜像构建过程默认不会挂载宿主机的GPU驱动文件（如/lib64/libcuda.so.1），而你手动进入容器时大概率使用了docker run --gpus ...这类NVIDIA Docker命令，该命令会自动将宿主机的GPU驱动库挂载到容器内，因此编译时能找到依赖。但构建阶段没有这个挂载操作，导致链接器找不到libcuda.so.1，触发报错。

方案1：基于NVIDIA CUDA镜像的多阶段构建（推荐） #

使用NVIDIA官方提供的CUDA开发镜像作为构建基础，这类镜像自带完整的CUDA开发环境与依赖库，无需依赖宿主机驱动，构建出的镜像兼容性和可移植性更强：

# 第一阶段：编译阶段
FROM nvidia/cuda:12.1.1-devel-ubuntu22.04 AS builder

# 安装编译依赖工具
RUN apt-get update && apt-get install -y --no-install-recommends git make gcc

# 克隆代码并完成编译
RUN git clone <你的代码仓库URL> && cd APPFolder && make

# 第二阶段：运行阶段（可选，用轻量runtime镜像减小体积）
FROM nvidia/cuda:12.1.1-runtime-ubuntu22.04

# 复制编译好的二进制文件到运行镜像
COPY --from=builder /path/to/your/compiled/app /usr/local/bin/

# 设置容器启动命令
CMD ["your-app-name"]

方案2：修改链接参数忽略警告（仅适用于运行时能挂载驱动的场景） #

如果该警告仅为链接阶段提示，且你确定运行容器时会通过--gpus参数挂载宿主机驱动，可以修改Makefile的链接参数，让链接器忽略未找到的共享库依赖：

修改Makefile中的LDFLAGS配置：

LDFLAGS += -Wl,--allow-shlib-undefined

然后在Dockerfile中正常执行编译：

FROM 你的基础镜像

RUN apt-get update && apt-get install -y git make gcc

RUN git clone <你的代码仓库URL> && cd APPFolder && make

注意：使用此方案时，运行容器必须加上--gpus all参数，否则运行时会因找不到libcuda.so.1而崩溃。

方案3：构建阶段手动挂载宿主机驱动（不推荐） #

若必须在构建阶段依赖宿主机驱动，可通过docker build的挂载参数将宿主机GPU驱动目录挂载到构建容器内：

构建命令：

docker build --mount type=bind,source=/usr/lib/nvidia,target=/usr/lib/nvidia -t your-image-name .

同时在Dockerfile中添加环境变量，让链接器能找到驱动库：

ENV LD_LIBRARY_PATH=/usr/lib/nvidia:$LD_LIBRARY_PATH

此方案缺点是镜像与宿主机驱动版本强绑定，移植性极差，仅适合特殊场景临时使用。

内容的提问来源于stack exchange，提问作者Yonatan