DrSnowbird/cuda-pytorch-docker
Nvidia CUDA for GPU + PyTorch (latest) in Docker
This project provides a pre-configured Docker container designed for AI/ML/DL scientists. It packages the latest NVIDIA CUDA and PyTorch, along with Jupyter Notebooks, making it ready-to-use for deep learning experiments. The container simplifies the often-complex setup of compatible deep learning environments, allowing users to focus on model development rather than installation issues.
No commits in the last 6 months.
Use this if you are an AI/ML/DL scientist needing a reliable, pre-configured environment for PyTorch and NVIDIA GPU computing, especially in corporate settings with SSL proxy challenges.
Not ideal if you have low-end GPU/CPU hardware or limited memory, as this container is resource-intensive and optimized for powerful deep learning setups.
Stars
7
Forks
2
Language
Jupyter Notebook
License
Apache-2.0
Category
Last pushed
Feb 29, 2024
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/DrSnowbird/cuda-pytorch-docker"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
replicate/cog
Containers for machine learning
dusty-nv/jetson-containers
Machine Learning Containers for NVIDIA Jetson and JetPack-L4T
rsnk96/Ubuntu-Setup-Scripts
Scripts to help you set up your Ubuntu quickly, especially if you're in any subfield of Data...
open-ce/open-ce
This repository provides the Open-CE environment files and version definitions for each Open-CE release.
lablup/backend.ai-kernels
Repository of Backend.AI-enabled container recipes