如何查看自己的tensorflow是不是gpu
时间: 2023-05-26 14:02:43 浏览: 107
tensorflow-gpu测试代码
5星 · 资源好评率100%
可以通过以下代码来查看自己的tensorflow是否支持GPU:
```python
import tensorflow as tf
if tf.test.gpu_device_name():
print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
print("Please install GPU version of TF")
```
如果输出结果中包含GPU设备的名称,则说明tensorflow支持GPU。如果输出结果中没有GPU设备的名称,则说明tensorflow不支持GPU。
阅读全文