判断CUDA是否可用：

tf.test.is_built_with_cuda()

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