setup_environment.py

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)