原文链接: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
)