.

Dockerfile

@@ -1,9 +1,19 @@
-# ベースイメージとしてPython 3.9を使用
-FROM python:3.9-slim
+# ベースイメージとしてCUDA対応のUbuntuを使用
+FROM nvidia/cuda:12.2.2-base-ubuntu20.04
 
 # 作業ディレクトリを設定
 WORKDIR /app
 
+# 必要なパッケージをインストール
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3-pip \
+    python3-dev \
+    && 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
 

app.py

@@ -210,12 +210,4 @@ with gr.Blocks() as demo:
     # JavaScriptの読み込み
     demo.load(js=script)
 
-# cudaのバージョンをprint
-import torch
-print("CUDA Version:", torch.version.cuda)
-
-# ubuntuのバージョンをprint
-import platform
-print("Ubuntu Version:", platform.linux_distribution())
-
 demo.launch()