官方文档:https://www.tensorflow.org/api_docs/python/tf/test/is_gpu_available
import tensorflow as tf
tf.test.is_gpu_available(
cuda_only=False,
min_cuda_compute_capability=None
)
若tensorflow_gpu成功安装,则输出显卡信息以及True。
官方文档:https://www.tensorflow.org/api_docs/python/tf/test/is_gpu_available
import tensorflow as tf
tf.test.is_gpu_available(
cuda_only=False,
min_cuda_compute_capability=None
)
若tensorflow_gpu成功安装,则输出显卡信息以及True。