检测tensorflow-gpu可用
时间: 2023-04-24 11:05:22 浏览: 672
TensorFlow-gpu测试
您可以通过以下步骤检测tensorflow-gpu是否可用:
1. 安装tensorflow-gpu:在命令行中输入“pip install tensorflow-gpu”进行安装。
2. 运行以下代码:
```
import tensorflow as tf
tf.test.is_gpu_available()
```
如果返回值为True,则说明tensorflow-gpu可用。如果返回值为False,则说明tensorflow-gpu不可用。
阅读全文