how to tell tensorflow to use gpu code example
Example 1: tensorflow check gpu
tf.config.list_physical_devices('GPU')
Example 2: tensorflow gpu test
tf.test.is_gpu_available(
cuda_only=False, min_cuda_compute_capability=None
)
tf.config.list_physical_devices('GPU')
tf.test.is_gpu_available(
cuda_only=False, min_cuda_compute_capability=None
)