Update app.py

app.py

@@ -28,3 +28,26 @@ if gpus:
 # TensorFlow 및 시스템 정보 확인
 print("TensorFlow 버전:", tf.__version__)
 print("사용 가능한 장치:", tf.config.list_physical_devices())
+
+import os
+import tensorflow as tf
+
+# oneDNN 최적화 비활성화
+os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
+
+# GPU 비활성화 (CUDA 문제 해결 시)
+os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
+
+# TensorFlow GPU 메모리 설정
+gpus = tf.config.experimental.list_physical_devices('GPU')
+if gpus:
+    try:
+        for gpu in gpus:
+            tf.config.experimental.set_memory_growth(gpu, True)
+        print("GPU 메모리 증가 허용 설정 완료")
+    except RuntimeError as e:
+        print(f"GPU 설정 오류: {e}")
+
+# TensorFlow 실행 확인
+print("TensorFlow 버전:", tf.__version__)
+print("사용 가능한 장치:", tf.config.list_physical_devices())