Top 6 Python nvidia-gpu Projects
-
modulus
Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods
-
InfluxDB
Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.
-
dream-factory
Multi-threaded GUI manager for mass creation of AI-generated art with support for multiple GPUs.
-
cubestat
Horizon chart for CPU/GPU/Neural Engine utilization monitoring on Apple M1/M2 and nVidia GPUs on Linux
I wouldn’t say ROCm code is “slower”, per se, but in practice that’s how it presents. References:
https://github.com/InternLM/lmdeploy
https://github.com/vllm-project/vllm
https://github.com/OpenNMT/CTranslate2
You know what’s missing from all of these and many more like them? Support for ROCm. This is all before you get to the really wildly performant stuff like Triton Inference Server, FasterTransformer, TensorRT-LLM, etc.
ROCm is at the “get it to work stage” (see top comment, blog posts everywhere celebrating minor successes, etc). CUDA is at the “wring every last penny of performance out of this thing” stage.
In terms of hardware support, I think that one is obvious. The U in CUDA originally stood for unified. Look at the list of chips supported by Nvidia drivers and CUDA releases. Literally anything from at least the past 10 years that has Nvidia printed on the box will just run CUDA code.
One of my projects specifically targets Pascal up - when I thought even Pascal was a stretch. Cue my surprise when I got a report of someone casually firing it up on Maxwell when I was pretty certain there was no way it could work.
A Maxwell laptop chip. It also runs just as well on an H100.
THAT is hardware support.
Project mention: 80's-90's fantasy art character generator w/ dynamic prompt (DS&D) | /r/StableDiffusion | 2023-05-12Good stuff! For anyone else that uses Dream Factory, I've converted this to a .prompts file for use with it here:
Python nvidia-gpu related posts
Index
What are some of the best open-source nvidia-gpu projects in Python? This list will help you:
Project | Stars | |
---|---|---|
1 | lmdeploy | 2,391 |
2 | modulus | 661 |
3 | dream-factory | 471 |
4 | genv | 415 |
5 | cubestat | 20 |
6 | k80-linux-cooling | 0 |
Sponsored