Create setup_environment.py

setup_environment.py

@@ -0,0 +1,12 @@
+import os
+import tensorflow as tf
+import logging
+
+# Disable oneDNN custom operations if needed
+os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
+
+# Disable GPU usage (force TensorFlow to use CPU)
+os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
+
+# Suppress TensorFlow warnings and logs
+tf.get_logger().setLevel(logging.ERROR)