测试tensorflow gpu是否安装成功
时间: 2023-04-24 12:01:00 浏览: 1450
TensorFlow-gpu测试
您好,如果您想测试TensorFlow GPU是否安装成功,可以使用以下代码:
```python
import tensorflow as tf
# 检查是否使用GPU
if tf.test.gpu_device_name():
print('GPU已安装')
else:
print('GPU未安装')
```
如果输出结果为“GPU已安装”,则说明TensorFlow GPU已经成功安装。如果输出结果为“GPU未安装”,则需要检查您的GPU驱动和TensorFlow版本是否匹配,以及是否正确安装了CUDA和cuDNN等依赖库。
阅读全文