.

Dockerfile

@@ -20,10 +20,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 COPY requirements.txt /app/requirements.txt
 RUN pip install --no-cache-dir -r requirements.txt
 
-# 環境変数を設定
-ENV PATH="/usr/local/cuda/bin:${PATH}"
-ENV XLA_FLAGS="--xla_gpu_cuda_data_dir=/usr/local/cuda"
-
 # アプリケーションコードをコンテナにコピー
 COPY . /app
 

docker-compose.yml

@@ -14,7 +14,3 @@ services:
             - capabilities: [ "gpu" ]
               count: 1
               driver: nvidia
-    runtime: nvidia
-    environment:
-      - NVIDIA_VISIBLE_DEVICES=all
-      - NVIDIA_DRIVER_CAPABILITIES=compute,utility

scripts/generate_prompt.py

@@ -7,20 +7,6 @@ from huggingface_hub import hf_hub_download
 
 import spaces
 
-import tensorflow as tf
-
-gpus = tf.config.list_physical_devices('GPU')
-if gpus:
-    try:
-        for gpu in gpus:
-            tf.config.experimental.set_memory_growth(gpu, True)
-        print("GPUs are available and memory growth is set.")
-    except RuntimeError as e:
-        print("Error setting GPU memory growth:", e)
-else:
-    print("No GPUs are available.")
-
-
 # 画像サイズの設定
 IMAGE_SIZE = 448