TensorFlow 怎么开启cpu加速
时间: 2024-04-30 16:22:55 浏览: 238
搭建TensorFlow(CPU版本)环境
TensorFlow 默认会使用 CPU 和 GPU 进行计算。如果需要强制 TensorFlow 只使用 CPU 进行计算,可以通过以下代码进行设置:
```python
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
```
这将设置 `CUDA_VISIBLE_DEVICES` 环境变量为 `-1`,强制 TensorFlow 只使用 CPU。在使用这个设置之后,TensorFlow 将不再尝试使用 GPU 进行计算,从而达到 CPU 加速的效果。
阅读全文