python - 如何仅使用普通 python 激活 google colab gpu

Question

我是 google colab 的新手。

我正在尝试在那里进行深度学习。

我编写了一个类来仅使用 python 创建和训练 LSTM 网络——而不是任何特定的深度学习库，如 tensorflow、pytorch 等。

我以为我使用的是 gpu，因为我在 colab 中正确选择了运行时类型。

然而，在代码执行期间，我有时会收到退出 gpu 模式的消息，因为我没有使用它。

所以，我的问题是:如何使用 google colab gpu，只使用简单的 python，而不使用特殊的 AI 库？是否有类似“装饰器代码”之类的东西可以放入我的原始代码中，以便 gpu 被激活？

Answer 1

使用 PyTorch 或 Tensorflow 等框架更容易。

如果没有，您可以尝试 pycuda 或 numba，它们更接近于“纯”GPU 编程。这比仅使用 PyTorch 还要难。