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

你好，我知道分析数据和使用人工智能的关键是使用gpu，而不是cpu。问题是我不知道如何在VisualStudio代码中与Python一起使用它，我使用Ubuntu，我已经安装了nvidia，您必须与设计用于GPU的库一起使用

在CUDA/ROC支持下，您可以使用Numba将Python代码直接编译为二进制代码，但我不知道它有多大的局限性

另一种方法是调用为并行计算（如OpenCL）设计的API，为此有PyOpenCL绑定

有一点限制，但有时是有效的方法-OpenGL/DirectX计算着色器，它们非常简单，但如果需要在小批量中来回传输数据，则速度不会太快