Python & CUDA のDockerfile

機械学習目的など、GPUを搭載したマシン上でCUDAとPythonを実行するためのDockerfileを示します。

Dockerfileはこちら!

PythonやCUDAのバージョンは適宜変更してください。

FROM nvidia/cuda:11.7.1-devel-ubuntu22.04

# timezone setting で止まる問題の対策
ENV DEBIAN_FRONTEND=noninteractive

# pythonや開発に必要なものをインストール
RUN apt update \
    && apt -y upgrade \
    && apt install -y software-properties-common tzdata \
    && add-apt-repository -y ppa:deadsnakes/ppa \
    && apt install -y python3.7 python3.7-distutils python3-pip \
    && apt install -y git wget curl git tmux vim

# パスを通す
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.7 1\
    && update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1

# pythonパッケージのバージョンアップ
RUN python -m pip install --upgrade pip wheel setuptools

docker-compose.yml

セットで使いそうなdocker-compose.ymlを以下に示します。

version: '3'

services:
  python_cuda:
    build:
      context: .
    image: python_cuda
    volumes:
      - ${PWD}:/workspace
    working_dir: /workspace
    shm_size: '4gb'
    tty: true
    privileged: true
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [ gpu ]

参考文献

• PyTorch + CUDA の Dockerfile