我目前正在研究java中的一个项目，它应该在几个GPU上运行python任务（小型机器学习任务）。 Java应该接受任务并尝试将其并行。
我知道我可以使用OpenCL \ CUDA在GPU上执行任务，我可以使用几个库来运行Java中的Python代码。
我现在看到的唯一方法是使用PyCUDA编写Python代码，然后从Java运行它。但我不确定是否可以通过Java在GPU上并行Python代码。因为只运行脚本与选择特别的向量和矩阵运算并在GPU上有效运行它们不同。
有没有图书馆可以做到这一点还是不可能？
答案 0 :(得分：1)
使用 TornadoVM [1]，您可以编写 Java 代码以通过 OpenCL 和/或 CUDA PTX 卸载到 GPU。并且可以从 GraalVM 之上的任何 PL 调用 Java 代码，包括 Python、Javascript、R 等 [2, 3]。