feat: rocm/dev-ubuntu-22.04でセットアップ

Author: itta611

Dockerfile

@@ -1,33 +1,27 @@
-# AMD GPU用のベースイメージに変更
-FROM rocm/pytorch:rocm6.4.4_ubuntu24.04_py3.12_pytorch_release_2.7.1
+# ROCm 6.4.4ベースのPyTorch環境
+FROM rocm/dev-ubuntu-22.04:6.4.4
 
-WORKDIR /app
+# 環境変数の設定
+ENV DEBIAN_FRONTEND=noninteractive
+ENV ROCM_VERSION=6.4.4
+ENV PATH=/opt/rocm/bin:$PATH
+ENV LD_LIBRARY_PATH=/opt/rocm/lib:$LD_LIBRARY_PATH
 
+# 基本パッケージのインストール
 RUN apt-get update && apt-get install -y \
-    libgl1-mesa-glx \
-    libglib2.0-0 \
-    libsm6 \
-    libxext6 \
-    libxrender-dev \
-    libgomp1 \
-    libglib2.0-0 \
-    libgl1-mesa-dev \
-    curl \
+    python3-pip \
+    python3-dev \
+    git \
+    wget \
+    vim \
     && rm -rf /var/lib/apt/lists/*
 
-COPY requirements.txt .
-# ROCm用のPyTorchを先にインストール（requirements.txt内のtorch/torchvisionを上書き）
-RUN pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/rocm6.2
-# その他の依存関係をインストール
-RUN pip install --no-cache-dir -r requirements.txt
+# pipのアップグレード
+RUN pip3 install --upgrade pip
 
-COPY main.py .
+# PyTorch ROCm版のインストール
+# ROCm 6.4.4に対応するPyTorch 2.5系をインストール
+RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2
 
-# AMD GPU環境変数
-ENV HSA_OVERRIDE_GFX_VERSION=10.3.0
-ENV HIP_VISIBLE_DEVICES=0
-ENV ROCR_VISIBLE_DEVICES=0
-
-EXPOSE 8000
-
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000", "--reload"]
+# 作業ディレクトリの設定
+WORKDIR /workspace