使用Conda环境创建Docker镜像的完整步骤指南

conda activate llama-factory
conda env export --name llama-factory > environment.yml

# 基于官方的 Conda 镜像
FROM continuumio/miniconda3

# 设置工作目录
WORKDIR /app

# 将环境配置文件复制到容器中
COPY environment.yml /app/

# 创建 Conda 环境并安装依赖
RUN conda env create -f environment.yml

# 激活环境
SHELL ["conda", "run", "-n", "llama-factory", "/bin/bash", "-c"]

# 设置工作目录（可选）
WORKDIR /app

# 容器启动时，默认激活环境并启动 bash
ENTRYPOINT ["conda", "run", "-n", "llama-factory", "/bin/bash"]

# 如果容器启动时需要执行其他命令，使用 CMD 指定
CMD ["bash"]

cp -r /home/lizhijun/anaconda3/envs/llama-factory ./llama-factory

COPY llama-factory-env /opt/conda/envs/llama-factory
COPY LLaMA-Factory /workspace/LLaMA-Factory

# 使用Ubuntu基础镜像
FROM ubuntu:20.04

# 避免交互式提示
ENV DEBIAN_FRONTEND=noninteractive

# 安装基础工具
RUN apt-get update && \
    apt-get install -y \
    wget \
    bzip2 \
    ca-certificates \
    libglib2.0-0 \
    libxext6 \
    libsm6 \
    libxrender1 \
    git \
    && rm -rf /var/lib/apt/lists/*

# 安装 Miniconda
RUN wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \
    bash ~/miniconda.sh -b -p /opt/conda && \
    rm ~/miniconda.sh && \
    /opt/conda/bin/conda clean -a -y

# 添加 Miniconda 到 PATH
ENV PATH /opt/conda/bin:$PATH

# 创建必要的目录
WORKDIR /workspace

# 从当前目录复制环境
COPY llama-factory-env /opt/conda/envs/llama-factory


# 确保复制的环境可用并升级 pip
RUN /bin/bash -c "conda init bash && \
    source ~/.bashrc && \
    source activate llama-factory && \
    python -m pip install --upgrade pip"


# 复制 LLaMA-Factory 文件夹
COPY LLaMA-Factory /workspace/LLaMA-Factory

# 进入 LLaMA-Factory 目录
WORKDIR /workspace/LLaMA-Factory

# 设置容器启动时默认使用的 Shell
CMD ["/bin/bash"]

docker build -t llama-factory-image .
docker run --gpus all -it llama-factory-image

docker save -o llama-factory-image.tar llama-factory-image
docker load -i llama-factory-image.tar