【Dockerfile】DockerにConda環境を構築し、仮想環境をActivateする

CUDAイメージ上に Miniconda or Anaconda 環境を構築し、conda or pip を仮想環境にインストール、アクティベートするところまでをDockerfileで完結させるためのDockerfileを記載します。

実現したいこと

• CUDAイメージ上にMiniconda(Anaconda)環境を構築
• Dockerfile内で、任意の仮想環境を作成し、パッケージをインストール
• Docker run (attach)した際に、Dockerfileに記載した仮想環境下に入る(conda activateする必要がない)

Dockerfile

Dockerfile

FROM nvidia/cuda:11.2.1-devel-ubuntu20.04

RUN apt-get update && apt-get install -y \
    sudo \
    wget \
    vim
WORKDIR /opt

RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
    sh Miniconda3-latest-Linux-x86_64.sh -b -p /opt/miniconda3 && \
    rm -r Miniconda3-latest-Linux-x86_64.sh

ENV PATH /opt/miniconda3/bin:$PATH

COPY <env_file_name>.yml .

RUN pip install --upgrade pip && \
    conda update -n base -c defaults conda && \
    conda env create -n <env_name> -f <env_file_name>.yml && \
    conda init && \
    echo "conda activate <env_name>" >> ~/.bashrc

ENV CONDA_DEFAULT_ENV <env_name> && \
    PATH /opt/conda/envs/<env_name>/bin:$PATH

WORKDIR /

CMD ["/bin/bash"]

NVIDIA Containers / Miniconda

NVIDIA CUDA Image 一覧はこちらから
Miniconda 一覧はこちらから