CPU和GPU Tensorflow安装

Question

我对TensorFlow很陌生。

我想在CUDA GPU上运行我的代码。因此，在安装普通的TensorFlow之后，我已经安装了TensorFlow -gpu。

我如何告诉Python它需要基于gpu的TensorFlow？

Answer 1

如果您安装了tensorflow-gpu，那么实际上也没有任何理由也需要tensorflow。没有gpu的话，它将直接在cpu上运行。

要具体使用哪个GPU：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

可以代替"0"列出GPU（如果有多个），也可以列出""（如果希望它在cpu上运行）。

或者，在会话中指定：

sess = tf.Session(config=tf.ConfigProto(device_count={'GPU': 0}))

此外，您可以通过打开python控制台并键入以下命令来检查计算机优先使用的版本：

>>> import tensorflow
>>> tensorflow
<module 'tensorflow' from 
'/home/.../python3.6/site- packages/tensorflow/__init__.py'>
                                       ^
                                       |
                                      here