在GPU上强制使用python脚本

Question

有没有办法在GPU上强制使用Python脚本？在我的代码中，我使用了tensorflow和keras，并且我已经有了tensorflow-gpu版本，但是我的代码仍然在CPU上运行。我想知道是否有一种方法可以强制在Tensorflow，Numpy或其他平台上独立运行GPU。

Answer 1

您的tensorflow-gpu安装可能存在问题。尝试创建一个单独的环境，然后仅重新安装tensorflow-gpu。

Similar post

Answer 2

对于TensorFlow（但通常不是python），这里有一个很好的说明：https://www.tensorflow.org/guide/gpu

要强制在特定处理器（CPU或GPU）上执行功能，请对tf.device（）使用TensorFlow调用，如下所示：

import tensorflow as tf
with tf.device('/GPU:0'):
  a = tf.constant([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
  b = tf.constant([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]])
  c = tf.matmul(a, b)

在这种情况下，a和b的数据将存储在GPU0上，并且“ matmul”操作也将在GPU0上执行。