anibali / docker-pytorch
A Docker image for PyTorch
☆982Updated last year
Alternatives and similar repositories for docker-pytorch:
Users that are interested in docker-pytorch are comparing it to the libraries listed below
- Profiling and inspecting memory in pytorch☆1,048Updated 7 months ago
- A Python module for getting the GPU status from NVIDA GPUs using nvidia-smi programmically in Python☆1,175Updated 11 months ago
- Standalone TFRecord reader/writer with PyTorch data loaders☆880Updated 7 months ago
- Tutorial for building a custom CUDA function for Pytorch☆512Updated 6 years ago
- A small package to create visualizations of PyTorch execution graphs☆3,320Updated 3 months ago
- A lightweight library for PyTorch training tools and utilities☆1,684Updated this week
- An unofficial styleguide and best practices summary for PyTorch☆1,970Updated 3 years ago
- Debug PyTorch code using PySnooper☆799Updated 3 years ago
- Experimental ground for optimizing memory of pytorch models☆366Updated 6 years ago
- A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.☆894Updated last year
- My best practice of training large dataset using PyTorch.☆1,093Updated 10 months ago
- Deep Learning Experiment Management☆640Updated 2 years ago
- Code snippets created for the PyTorch discussion board☆561Updated 4 years ago
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,819Updated last year
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,598Updated this week
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆685Updated 5 years ago
- PyTorch Boilerplate For Research☆609Updated 3 years ago
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,230Updated this week
- Train AI models efficiently on medical images using any framework☆1,872Updated 9 months ago
- Estimates the size of a PyTorch model in memory☆357Updated 4 years ago
- PyTorch layer-by-layer model profiler☆606Updated 3 years ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,037Updated last year
- Library for 8-bit optimizers and quantization routines.☆717Updated 2 years ago
- View model summaries in PyTorch!☆2,736Updated this week
- The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflo…☆804Updated 2 years ago
- Unit Testing for pytorch, based on mltest☆310Updated 4 years ago
- A learning rate range test implementation in PyTorch☆950Updated 3 months ago
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,712Updated this week
- ☆261Updated 5 years ago
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,718Updated 3 weeks ago