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

Question

您好，我知道分析數據和使用人工智能的關鍵是使用 gpu 而不是 cpu。 問題是我不知道如何在Visual Studio代碼中將它與Python一起使用，我使用Ubuntu，我已經安裝了nvidia

Answer 1

您必須使用旨在與 GPU 一起使用的庫。

您可以使用 Numba 將 Python 代碼直接編譯為具有 CUDA/ROC 支持的二進制文件，但我真的不知道它有多大限制。

另一種方法是調用為並行計算設計的 API，例如 OpenCL，PyOpenCL 對此進行綁定。

有點限制，但有時是有效的方式 - OpenGL/DirectX 計算着色器，它們非常簡單，但如果您需要小批量來回傳輸數據，它們就不會那么快。