TensorFlow:判断CUDA和GPU是否可用

原创
2018/05/17 12:56
阅读数 6.9K

无偿推广: Mac OS 最强鼠标改键软件:BetterAndBetter

判断CUDA是否可用:

tf.test.is_built_with_cuda()

参考: tf.test.is_built_with_cuda | TensorFlow


判断GPU是否可用:

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

参考: tf.test.is_gpu_available | TensorFlow

展开阅读全文
打赏
0
0 收藏
分享
加载中
更多评论
打赏
0 评论
0 收藏
0
分享
返回顶部
顶部