在Nvidia/CUDA Runtime Docker镜像中安装Micromamba失败求助

问题核心在于Docker的每个RUN指令都是独立的shell进程，前序RUN中设置的环境变量和shell钩子无法在后续RUN中保留，导致micromamba activate无法找到必要的配置。以下是两种可行的解决方式：

方法1：合并命令到单个RUN进程 #

将Micromamba的安装、初始化、激活操作放在同一个RUN指令中，确保所有操作共享同一shell环境：

ARG CUDA=11.1.1
FROM nvidia/cuda:${CUDA}-cudnn8-runtime-ubuntu18.04
ARG CUDA

# 安装基础依赖工具
RUN apt-get update && apt-get install -y curl && rm -rf /var/lib/apt/lists/*

# 安装并激活Micromamba（同进程执行所有操作）
RUN curl -Ls https://micro.mamba.pm/api/micromamba/linux-64/latest | tar -xvj bin/micromamba && \
    export MAMBA_ROOT_PREFIX=./micromamba && \
    eval "$(./bin/micromamba shell hook -s bash)" && \
    micromamba activate && \
    # 在此添加你的包安装命令，例如：micromamba install -y python=3.8
    echo "Micromamba环境激活成功"

方法2：初始化shell配置，让后续进程自动加载钩子 #

通过micromamba shell init将钩子写入bash配置文件，后续RUN中加载配置即可直接激活：

ARG CUDA=11.1.1
FROM nvidia/cuda:${CUDA}-cudnn8-runtime-ubuntu18.04
ARG CUDA

RUN apt-get update && apt-get install -y curl bash && rm -rf /var/lib/apt/lists/*

# 安装Micromamba并初始化shell钩子
RUN curl -Ls https://micro.mamba.pm/api/micromamba/linux-64/latest | tar -xvj bin/micromamba && \
    ./bin/micromamba shell init --shell=bash --prefix=./micromamba

# 后续RUN加载配置后直接激活
RUN source ~/.bashrc && micromamba activate && \
    micromamba install -y numpy

额外优化建议 #

• 使用绝对路径配置MAMBA_ROOT_PREFIX（如/opt/micromamba），避免相对路径带来的路径解析问题
• 可以通过ENV指令设置全局环境变量，简化后续Micromamba命令调用：

  ENV MAMBA_EXE=/bin/micromamba
  ENV MAMBA_ROOT_PREFIX=/opt/micromamba
  

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