# ベースイメージの指定
FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04

# ユーザー名をビルド引数として定義
ARG USER_NAME=tadokuno

# 必要なパッケージのインストール
RUN apt update && apt install -y sudo vim net-tools git python3-pip software-properties-common

# ユーザーの追加
RUN adduser --disabled-password --gecos '' ${USER_NAME} && \
    echo "${USER_NAME} ALL=(ALL) NOPASSWD: ALL" > /etc/sudoers.d/${USER_NAME} && \
    chmod 0440 /etc/sudoers.d/${USER_NAME}

# Pythonのインストール
RUN add-apt-repository ppa:deadsnakes/ppa && \
    apt update && \
    apt install -y python3.13 python3.13-venv python3.10 python3.10-venv && \
    update-alternatives --install /usr/bin/python python /usr/bin/python3.13 100 && \
    update-alternatives --install /usr/bin/python python /usr/bin/python3.10 110

# CUDA環境変数スクリプトを.bashrcに直接追加
RUN echo '\n\
if [ -e /usr/local/cuda ]; then\n\
    export PATH="/home/${USER_NAME}/.local/bin:/usr/local/cuda/bin:$PATH"\n\
    export LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/TensorRT/lib:/opt/nvidia/nsight-compute/2023.1.1/host/linux-desktop-glibc_2_11_3-x64/Mesa"\n\
fi\n' >> /home/${USER_NAME}/.bashrc

# ユーザーに切り替え
USER ${USER_NAME}
WORKDIR /home/${USER_NAME}

# コンテナ起動時にbashを実行
CMD ["bash"]