python - Arrayfire python支持多GPU编程吗

Question

我正在尝试使用 arrayFire python ( https://github.com/arrayfire/arrayfire-python ) 进行多 GPU 编程。

但是，当我尝试将其与并发 futures ( https://docs.python.org/3/library/concurrent.futures.html ) 库连接时，我遇到了同步问题。

有人知道如何使用 arrayfire-python 在多个 GPU 上并行处理吗？

Answer 1

ArrayFire 允许多 GPU 编程，但不会自动分配工作负载。由用户决定哪些内存和功能在哪个设备上运行。

ArrayFire 目前的情况不是线程安全的。因此，在多个线程上运行任何内容都可能会导致问题。

披露:我是 ArrayFire 的开发人员。