wookayin / gpustat
π A simple command-line utility for querying and monitoring GPU status
β4,009Updated last month
Related projects: β
- GPU & Accelerator process monitoring for AMD, Apple, Huawei, Intel, NVIDIA and Qualcommβ7,981Updated 2 weeks ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorchβ8,303Updated 2 weeks ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)β8,362Updated this week
- NumPy & SciPy for GPUβ8,124Updated this week
- A small package to create visualizations of PyTorch execution graphsβ3,168Updated 5 months ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)β7,852Updated 2 weeks ago
- An interactive NVIDIA-GPU process viewer and beyond, the one-stop solution for GPU process management.β4,617Updated last week
- Model summary in PyTorch similar to `model.summary()` in Kerasβ4,001Updated 6 months ago
- A Python module for getting the GPU status from NVIDA GPUs using nvidia-smi programmically in Pythonβ1,120Updated 5 months ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.β4,512Updated this week
- A scikit-learn compatible neural network library that wraps PyTorchβ5,801Updated last week
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.β14,158Updated 2 weeks ago
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learβ¦β5,065Updated this week
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125β14,031Updated this week
- View model summaries in PyTorch!β2,499Updated last week
- Model interpretability and understanding for PyTorchβ4,810Updated this week
- torch-optimizer -- collection of optimizers for Pytorchβ3,012Updated 5 months ago
- cuML - RAPIDS Machine Learning Libraryβ4,158Updated this week
- A hyperparameter optimization frameworkβ10,527Updated last week
- Build and run Docker containers leveraging NVIDIA GPUsβ17,189Updated 9 months ago
- Hydra is a framework for elegantly configuring complex applicationsβ8,612Updated this week
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.β22,236Updated this week
- Code for visualizing the loss landscape of neural netsβ2,770Updated 2 years ago
- PyTorch deep learning projects made easy.β4,700Updated 3 months ago
- Sacred is a tool to help you configure, organize, log and reproduce experiments developed at IDSIA.β4,214Updated 3 weeks ago
- PyTorch extensions for high performance and large scale training.β3,149Updated 2 weeks ago
- A curated list of awesome self-supervised methodsβ6,113Updated 2 months ago
- Image augmentation for machine learning experiments.β14,343Updated last month
- Multi-GPU CUDA stress testβ1,345Updated 3 weeks ago
- Geometric Computer Vision Library for Spatial AIβ9,806Updated this week