Anaconda で PyTorch を使う Dockerfile

画像処理系の深層学習用．

Dockerfile

#----使用するcudaのバージョンとubuntuのバージョンを指定------
FROM nvidia/cuda:11.2.0-devel-ubuntu18.04
#-----------------------------------------------------

WORKDIR /root

# インタラクティブをオフ（イメージビルドが止まらないように）
ENV DEBIAN_FRONTEND=noninteractive

# Ubuntu のアップデート
RUN apt-get update

# Ubuntu のタイムゾーンの設定（一部ソフトのインストールに必要）
RUN apt-get install -y tzdata
ENV TZ=Asia/Tokyo

# 必要なソフトウェアの追加(Python3 を使う場合)
RUN apt-get install -y git vim gcc build-essential libssl-dev zlib1g-dev \ 
                        libbz2-dev libreadline-dev libsqlite3-dev \ 
                        wget curl llvm libncurses5-dev libncursesw5-dev \ 
                        xz-utils libffi-dev liblzma-dev libjpeg-dev \
                        python3 python3-pip


# OpenCV に必要なソフトの追加
RUN apt-get install -y libgl1-mesa-dev libglib2.0-0 libsm6 libxrender1 libxext6


#########################################
# anaconda 環境の設定
#########################################
ARG conda_ver="2022.10"
RUN set -x && \
    wget https://repo.anaconda.com/archive/Anaconda3-${conda_ver}-Linux-x86_64.sh && \
    bash Anaconda3-${conda_ver}-Linux-x86_64.sh -b && \
    rm Anaconda3-${conda_ver}-Linux-x86_64.sh
ENV PATH $PATH:/root/anaconda3/bin
RUN conda init bash
RUN conda update conda

# Python 3.9を使いたい場合
RUN conda create -n myenv python=3.9
RUN echo "conda activate myenv" >> ~/.bashrc
ENV CONDA_DEFAULT_ENV myenv && \
    PATH /root/conda/envs/myenv/bin:$PATH
SHELL ["conda", "run", "-n", "myenv", "/bin/bash", "-c"]

RUN pip install --upgrade pip && \
    pip install torch \
                torchvision

# $ python3 →純正のPython3
# $ python →AnacondaのPythonなので注意