update client code

Flower_v1/client_2.py

@@ -9,18 +9,24 @@ import numpy as np
 import sys
 import random
 
+# list up
 physical_devices = tf.config.list_physical_devices('GPU')
-if physical_devices:
+
+if len(physical_devices) > 0:
+    # Only use 1st GPU
+    tf.config.set_visible_devices(physical_devices[0], 'GPU')
+
+    # Avoid over mem
     tf.config.experimental.set_memory_growth(physical_devices[0], True)
-    print("GPU detected and memory growth enabled.")
+    print("GPU detected and memory growth enabled. Using GPU:", physical_devices[0].name)
 else:
     print("No GPU detected, using CPU.")
     
 # Set random seed for reproducibility
 def set_random_seed(seed):
-    random.seed(seed)  # Python random module seed
-    np.random.seed(seed)  # Numpy random seed
-    tf.random.set_seed(seed)  # TensorFlow random seed
+    random.seed(seed)  
+    np.random.seed(seed)  
+    tf.random.set_seed(seed) 
 
 # Set a specific seed value
 set_random_seed(42)