File size: 187 Bytes
748b6ee
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
import tensorflow as tf

# Check if GPU is available
if tf.test.is_gpu_available():
    print("GPU is available")
else:
    print("GPU is not available")

# Rest of your code...