File size: 254 Bytes
8d4d3e5
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
import tensorflow as tf

gpus = tf.config.list_physical_devices('GPU')
if gpus:
    print(f"Detected {len(gpus)} GPU(s):")
    for gpu in gpus:
        print(f"  - {gpu.name}")
else:
    print("No GPUs detected by TensorFlow. Check your configuration.")