test

Dockerfile

@@ -1,32 +1,11 @@
 # CUDA 12.1ベースのUbuntuイメージを使用
 # FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu20.04
-FROM ubuntu:20.04
+FROM python:3.10-slim
 
 # 環境変数の設定
 ENV DEBIAN_FRONTEND=noninteractive
 ENV TZ=Asia/Tokyo
 
-# システムの更新とPython 3.10のインストール
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    software-properties-common \
-    wget \
-    libopencv-dev \
-    && add-apt-repository ppa:deadsnakes/ppa \
-    && apt-get update && apt-get install -y --no-install-recommends \
-    python3.10 \
-    python3.10-dev \
-    python3.10-distutils \
-    && ln -sf /usr/share/zoneinfo/Asia/Tokyo /etc/localtime \
-    && apt-get clean \
-    && rm -rf /var/lib/apt/lists/*
-
-# pipのインストールとPythonのセットアップ
-RUN wget https://bootstrap.pypa.io/get-pip.py \
-    && python3.10 get-pip.py \
-    && rm get-pip.py \
-    && update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1 \
-    && update-alternatives --install /usr/bin/pip pip /usr/local/bin/pip3.10 1
-
 WORKDIR /app
 
 # 依存関係をインストール

requirements.txt

@@ -1,4 +1,4 @@
---extra-index-url https://download.pytorch.org/whl/cu116
+--extra-index-url https://download.pytorch.org/whl/cu121
 torch==2.2.0
 torchvision==0.17.0
 torchaudio==2.2.0