Upgrade to Pro — share decks privately, control downloads, hide ads and more …

Zero Code Change GPU Acceleration

Zero Code Change GPU Acceleration

Jacob Tomlinson

August 28, 2024
Tweet

More Decks by Jacob Tomlinson

Other Decks in Technology

Transcript

  1. Accelerated NetworkX Zero code change acceleration for NetworkX, powered by

    cuGraph • Zero code change GPU-acceleration for NetworkX code • Accelerates algorithms up to 600x, based on algorithm and graph size • Support for 60 popular graph algorithms and growing • Fallback to using CPU NetworkX for any unsupported algorithms NetworkX 3.2, CPU: Intel(R) Xeon(R) Platinum 8480CL 2TB, GPU: NVIDIA H100 80GB pip install nx-cugraph-cu12 --extra-index-url https://pypi.nvidia.com conda install -c rapidsai -c conda-forge -c nvidia nx-cugraph