查看tensorflow是否可以调用gpu

进入python编译环境，输入一下代码，如果结果是True，表示GPU可用

import tensorflow as tf
print(tf.test.is_gpu_available())

 

posted @ 2020-08-04 12:25  牛郎  阅读(9015)  评论(0)  编辑  收藏  举报