代码之家 › 专栏 › 技术社区 › Ronnypetson Souza da Silva

在使用GPU之前,是否有必要在Google Colaboratory上安装GPU库?

placement device gpu

1

Ronnypetson Souza da Silva · 技术社区 · 8 年前

我一直试图在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的矩阵乘法次数,并对它们进行了比较。

1 回复 | 直到 8 年前

1

0

RomRoc 8 年前

不,没有必要。在Colaboratory中,您应该检查运行时->更改运行时类型参数Hardware accelerator is GPU。

然后,为了测试Tensorflow是否使用它,您可以看到这个有趣的示例,它对我很有用: https://stackoverflow.com/a/43703735/9250875