tensorflow 设置调用gpu
时间: 2023-06-06 09:07:51 浏览: 99
编译好的 tensorflow2.8.0 c++接口库,支持gpu
可以使用以下代码设置在TensorFlow中调用GPU:
```python
import tensorflow as tf
# 设置TensorFlow使用GPU
physical_devices = tf.config.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(physical_devices[0], True)
```
这将启用TensorFlow在GPU上执行计算。
阅读全文