用tensorflow,请你写一段Python代码,看下gpu是否被调用
时间: 2024-10-17 10:05:07 浏览: 23
在tensorflow中设置使用某一块GPU、多GPU、CPU的操作
5星 · 资源好评率100%
在TensorFlow中检查GPU是否被使用的代码通常涉及到硬件环境检测。以下是一个简单的示例,它首先尝试在GPU上运行一个矩阵乘法操作,如果成功,则说明GPU被启用;如果失败,那么可能会提示GPU未启用或没有正确配置。
```python
import tensorflow as tf
from tensorflow.python.client import device_lib
def is_gpu_available():
# 获取可用设备信息
local_device_protos = device_lib.list_local_devices()
# 检查是否有GPU设备
return any(device.name.lower().startswith('device:gpu') for device in local_device_protos)
try:
with tf.device('/GPU:0'): # 如果有GPU,这行代码会在GPU上运行
a = tf.constant([[1.0, 2.0], [3.0, 4.0]])
b = tf.constant([[5.0, 6.0], [7.0, 8.0]])
c = tf.matmul(a, b)
except ValueError:
print("No GPU detected or not available.")
else:
if is_gpu_available():
print("Using GPU")
else:
print("CPU used instead of GPU")
阅读全文