同时安装Tensorflow和Tensorflow-gpu时如何使用Tensorflow-gpu？

Question

我在同一虚拟环境中同时安装了TF和TF-GPU。当我在iPython终端中对其进行测试时，它表明使用了TensorFlow，而不是TensorFlow-GPU。但是，我希望使用GPU版本。

该怎么做？我不应该将两者都安装在同一环境中吗？谢谢。

Answer 1

我可以推荐：

pip uninstall tensorflow
pip uninstall tensorflow-gpu
pip install tensorflow
pip install tensorflow-gpu

希望我知道更好的答案