查看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)
print(tf.__version__)
tf.test.is_built_with_cuda()
tf.test.is_gpu_available(cuda_only=False, min_cuda_compute_capability=None)