NVIDIA CUDA
PyCUDA
Caffe
wangx0102
V2EX  ›  CUDA

如何实现 CUDA 的分布式并行运算?

  •  
  •   wangx0102 · Aug 6, 2021 · 4444 views
    This topic created in 1744 days ago, the information mentioned may be changed or developed.

    导师给了一个程序,实现了一个中间件可以实现 CPU 和 GPU 运算的负载均衡。

    我的初步想法是把 CUDA 程序打包成 exe 或者.so 啥的,然后用 Python 调用,使用 Celery 实现分布式集群。

    希望大家能有更好的想法

    1 replies    2023-05-03 13:40:08 +08:00
    alqaz
        1
    alqaz  
       May 3, 2023
    框架的话建议用 Ray
    About   ·   Help   ·   Advertise   ·   Blog   ·   API   ·   FAQ   ·   Solana   ·   2957 Online   Highest 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 29ms · UTC 05:31 · PVG 13:31 · LAX 22:31 · JFK 01:31
    ♥ Do have faith in what you're doing.