我如何在 python Visual Studio 代码中使用我的 GPU

Question

您好，我知道分析数据和使用人工智能的关键是使用 gpu 而不是 cpu。 问题是我不知道如何在Visual Studio代码中将它与Python一起使用，我使用Ubuntu，我已经安装了nvidia

Answer 1

您必须使用旨在与 GPU 一起使用的库。

您可以使用 Numba 将 Python 代码直接编译为具有 CUDA/ROC 支持的二进制文件，但我真的不知道它有多大限制。

另一种方法是调用为并行计算设计的 API，例如 OpenCL，PyOpenCL 对此进行绑定。

有点限制，但有时是有效的方式 - OpenGL/DirectX 计算着色器，它们非常简单，但如果您需要小批量来回传输数据，它们就不会那么快。