判断CUDA是否可用：

tf.test.is_built_with_cuda()

posted @ 2020-04-07 11:53  wangaolin  阅读(3106)  评论(0)  编辑  收藏  举报