← all repositories

tlkh/ai-lab

All-in-one Docker container providing GPU-accelerated ML and deep learning frameworks for rapid prototyping.

436 stars JavaScript ML FrameworksData Tooling
ai-lab
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

This repository provides a pre-configured Docker image for AI and machine learning development. It bundles popular frameworks including PyTorch and TensorFlow alongside CUDA support for GPU acceleration, Jupyter notebooks for interactive development, and development tools like VSCode. The container runs on any system with NVIDIA drivers and nvidia-docker, offering a portable alternative to cloud-based deep learning VMs.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.