AutoUpdate TensorFlow in export.py

[glenn-jocher]

May help resolve #9375

Signed-off-by: Glenn Jocher glenn.jocher@ultralytics.com

🛠️ PR Summary

_{Made with ❤️ by Ultralytics Actions}

🌟 Summary

Improved TensorFlow export compatibility for YOLOv5 on different hardware.

📊 Key Changes

• Added a check to determine if TensorFlow or TensorFlow-CPU should be used based on CUDA availability.

🎯 Purpose & Impact

• 🎯 Ensures the export script uses the correct TensorFlow package depending on the user's system (with or without GPU support), enhancing compatibility and user experience.
• 🚀 Potentially reduces setup complexity and errors for users exporting YOLOv5 models to TensorFlow SavedModel format, as the script now automatically selects the appropriate TensorFlow dependency.