输入代码检验安装的tensorflow-gpu版本是否可以使用
时间: 2024-09-27 18:13:53 浏览: 33
在Python中,你可以通过运行特定的命令或者导入TensorFlow库并检查其版本信息来验证已经安装的TensorFlow-GPU版本。这里是一个简单的例子:
```python
# 首先尝试导入TensorFlow GPU模块
try:
import tensorflow as tf
from tensorflow.python.platform import gfile
# 检查GPU可用性
print("TensorFlow version:", tf.__version__)
if 'GPU' in tf.config.list_physical_devices('GPU'):
print("Found GPU!")
else:
print("No GPU found.")
except ImportError:
print("Error importing TensorFlow. Is it installed with GPU support?")
# 如果安装了GPU版本,还可以查看具体的GPU驱动和CUDA版本
if "GPU" in tf.config.list_physical_devices():
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
```
这段代码首先尝试导入TensorFlow,如果成功,它会打印出当前的TensorFlow版本以及是否存在GPU。如果找不到GPU,则会显示相应的提示。
阅读全文