tensorflow - Pycharm 中的 TensorFlow GPU 版本而不是 CPU 版本

Question

如何在 PyCharm（或其他环境）中使用 TensorFlow GPU 版本而不是 CPU 版本？

标签： tensorflowgpu

Answer 1

默认情况下，从 TensorFlow 2.1 开始（无论 PyCharm 或您正在编码的任何环境），TensorFlow 都将 CPU+GPU 包安装在一起。对于旧版本（我不推荐），您可以显式使用pip install tensorflow（旧 CPU 版本）或pip install tensorflow-gpu

如果要使用 CPU 而不是 GPU，可以执行以下操作：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '-1'
import tensorflow as tf //tf now uses CPU

这样，你可以强制TF只使用CPU，即使你已经可以正常使用GPU。