import tensorflow as tf
print(tf.test.is_gpu_available())
如果输出True，则可以应用，如果输出False则不能应用
【Tensorflow 2】查看GPU是否能应用
关注
打赏
import tensorflow as tf
print(tf.test.is_gpu_available())
如果输出True，则可以应用，如果输出False则不能应用
微信扫码登录