TensorFlow：判断CUDA和GPU是否可用

原文链接：https://blog.csdn.net/qq_45616304/article/details/112758607

查看tf版本

print(tf.__version__)

判断CUDA是否可用：

tf.test.is_built_with_cuda()

判断GPU是否可用：

 tf.test.is_gpu_available(
        cuda_only=False,
        min_cuda_compute_capability=None
    )