TensorFlow/gpu.py

3 lines
65 B
Python

import tensorflow as tf
print(tf.config.list_physical_devices())