python的tensorflow检查GPU可用性的代码
时间: 2023-06-08 11:04:14 浏览: 167
查看tensorflow版本号、查看cuda版本、查看cudnn版本、查看GPU可用性、查看cuda可用性.pdf
5星 · 资源好评率100%
以下是检查TensorFlow GPU可用性的Python代码:
```python
import tensorflow as tf
if tf.test.is_gpu_available():
print("GPU is available")
else:
print("GPU is not available")
```
这段代码将检查是否有可用的GPU。 如果GPU可用,代码将输出“GPU is available”;如果没有可用的GPU,它将输出“GPU is not available”。
阅读全文