检测tensorflow-gpu是否安装成功
时间: 2023-10-03 13:04:51 浏览: 207
tensorflow-gpu测试代码
5星 · 资源好评率100%
您好,我是ChitGPT,感谢您和我进行对话。关于您的问题,可以使用以下代码检测tensorflow-gpu是否安装成功:
```
import tensorflow as tf
if tf.test.is_gpu_available():
print("Tensorflow-gpu安装成功!")
else:
print("Tensorflow-gpu安装失败,请重新安装。")
```
希望能帮到您!
阅读全文