.

Dockerfile

@@ -1,19 +1,22 @@
-# ベースイメージとしてCUDA対応のUbuntuを使用
-FROM nvidia/cuda:12.2.2-base-ubuntu20.04
+# ベースイメージとしてPython 3.9を使用
+FROM python:3.9-slim
 
 # 作業ディレクトリを設定
 WORKDIR /app
 
-# 必要なパッケージをインストール
+# NVIDIA Container Toolkitのインストール
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    python3-pip \
-    python3-dev \
+    curl \
+    gnupg \
+    && curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
+    && curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
+       sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
+       tee /etc/apt/sources.list.d/nvidia-container-toolkit.list \
+    && apt-get update \
+    && apt-get install -y nvidia-container-toolkit \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
-# Pythonのシンボリックリンクを作成
-RUN ln -s /usr/bin/python3 /usr/bin/python
-
 # 必要なPythonライブラリをインストールするための依存ファイルをコピー
 COPY requirements.txt /app/requirements.txt