OutofAi/cudacanvas
Python Module for PyTorch Tensor Visualisation in CUDA Eliminating CPU Transfer
This tool helps machine learning scientists and engineers visualize images directly from PyTorch tensors during model training or evaluation. It takes raw image data stored as PyTorch tensors on a CUDA-enabled GPU and displays them in a window, making it easy to monitor image generation or processing in real-time. This is ideal for those working with deep learning models, especially in computer vision.
No commits in the last 6 months. Available on PyPI.
Use this if you are a machine learning scientist or engineer training PyTorch models on a GPU and need to see image outputs immediately without slowing down your workflow.
Not ideal if you primarily work with CPU-based PyTorch models or require advanced image plotting and analysis features beyond simple display.
Stars
41
Forks
3
Language
Python
License
MIT
Category
Last pushed
Jan 30, 2025
Commits (30d)
0
Dependencies
1
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/OutofAi/cudacanvas"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
brucefan1983/GPUMD
Graphics Processing Units Molecular Dynamics
iree-org/iree
A retargetable MLIR-based machine learning compiler and runtime toolkit.
uxlfoundation/oneDAL
oneAPI Data Analytics Library (oneDAL)
rapidsai/cuml
cuML - RAPIDS Machine Learning Library
NVIDIA/cutlass
CUDA Templates and Python DSLs for High-Performance Linear Algebra