Google Colaboratory 使用 GPU 之前需要安装 GPU 库吗？

Question

我一直在尝试在 Colaboratory 上将 GPU 与 tensorflow 结合使用，但是当我这样做时

a = tf.constant(np.random.rand(1000,20000))
b = tf.constant(np.random.rand(20000,1000))
with tf.device('/device:GPU:0'):
    c_gpu = tf.matmul(a,b)
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
print(sess.run(c_gpu))

操作的设备没有打印出来，虽然操作的结果是。我怀疑它没有使用 GPU，因为我已经测量了 GPU 和 CPU 的矩阵乘法时间并比较了它们。

Answer 1

不，没有必要。在 Colaboratory 中，您应该检查 Runtime -> Change runtime type 参数 Hardware accelerator 是否为 GPU。

然后测试 Tensorflow 是否使用它，你可以看到这个有趣的示例，它对我有用：

Google Colaboratory 使用 GPU 之前需要安装 GPU 库吗？

Is it necessary to install GPU libraries on Google Colaboratory before using GPU?

gpu

device

placement