gpu - 你可以在没有任何库的情况下在 Google Colab 中使用 GPU 吗？

Question

我已经用 Python 从头开始​​编写了一个神经网络，我正在使用 Google Colaboratory 来训练它。但是，如果我启用 GPU 或 TPU 加速，训练并不会更快。

当你在网上搜索示例时，它们都使用 Tensorflow 和其他库，并且使用 GPU 的训练时间比不使用 GPU 的时间短。

我做得对还是我错过了一些东西并且没有使用 GPU？

标签： gpugoogle-colaboratory

Answer 1

仅启用 GPU 或 TPU 不会解决您的问题，如果您不使用任何框架或库，则需要显式编码它们以在 GPU 上运行。