memo

dockermemo

# ==========================================================
# ベース: PyTorch 1.13.0 + CUDA 11.6 + cuDNN 8 (安定構成)
# ==========================================================
FROM pytorch/pytorch:1.13.0-cuda11.6-cudnn8-devel

# 非対話環境（遠隔操作対応）
ENV DEBIAN_FRONTEND=noninteractive
ENV PYTHONUNBUFFERED=1

# 基本ツール
RUN apt-get update && apt-get install -y --no-install-recommends \
    git wget curl build-essential cmake ninja-build libgl1-mesa-glx ffmpeg ca-certificates \
    && rm -rf /var/lib/apt/lists/*

# pip更新
RUN pip install --upgrade pip==23.2.1 setuptools==68.0.0 wheel==0.41.2

# ===============================================
# 基本ライブラリ（安定動作確認済みバージョン）
# ===============================================
RUN pip install \
    numpy==1.23.5 \
    scipy==1.10.1 \
    pandas==2.1.2 \
    opencv-python-headless==4.9.0.72 \
    matplotlib==3.7.1 \
    tqdm==4.65.0 \
    pillow==10.0.0 \
    pyyaml==6.0.1 \
    tensorboard==2.14.0 \
    networkx==2.8.8

# ===============================================
# spconv (sparse convolutionライブラリ)
# ===============================================
# CUDA 11.6 用の安定版バイナリ
RUN pip install cumm-cu116==2.1.3 spconv-cu116==2.3.1

# ===============================================
# OpenMMLab 関連 (MMDetection3D)
# ===============================================
RUN pip install openmim==0.4.0 && \
    mim install "mmengine>=0.7.1,<1.0.0" && \
    mim install "mmcv>=2.0.0rc4,<2.2.0" && \
    mim install "mmdet>=3.0.0,<3.3.0" && \
    mim install "mmdet3d==1.4.0"

# ===============================================
# 動作確認（ロードテスト）
# ===============================================
RUN python -c "import torch, mmcv, mmdet, mmdet3d; print('✅ MMDetection3D environment OK')"

# キャッシュ削除で軽量化
RUN rm -rf /root/.cache/pip