通过tensorflow判断当前环境是否支持使用GPU

方法一:在控制台或ipython环境输入如下代码

import tensorflow as tf
tf.test.is_gpu_available()

如下图所示,当前环境支持使用GPU

方法二:在控制台或ipython环境输入如下代码

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

如下图所示,输出了相关信息,表明当前环境支持使用GPU 

 

猜你喜欢

转载自blog.csdn.net/sinat_29217765/article/details/114919231