检测tensorFlow的GPU版本安装成功方法
时间: 2024-05-02 14:20:41 浏览: 70
TensorFlow-gpu测试
查看TensorFlow的安装版本:`pip show tensorflow`
如果tensorflow显示为gpu版本,则已经安装成功。
如果仍无法确定,可以尝试在Python中导入TensorFlow并检查是否正在使用GPU。```python
import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))
```如果已安装GPU版本,输出应该大于0。
阅读全文