GitHub / suvash / nixos-nvidia-cuda-python-docker-compose
A step-by-step guide to setting up Nvidia GPUs with CUDA support running on Docker (and Compose) containers on NixOS host
JSON API: http://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/suvash%2Fnixos-nvidia-cuda-python-docker-compose
PURL: pkg:github/suvash/nixos-nvidia-cuda-python-docker-compose
Stars: 41
Forks: 5
Open issues: 0
License: None
Language: Dockerfile
Size: 69.3 KB
Dependencies parsed at:
24
Created at: almost 4 years ago
Updated at: 3 months ago
Pushed at: 4 months ago
Last synced at: 3 months ago
Topics: deep-learning, docker-compose, jax, nixos, nvidia, nvidia-cuda, nvidia-docker, nvidia-smi, python, pytorch, tensorflow, triton
Funding Links https://github.com/sponsors/suvash
- torch ==1.11.0
- torchaudio ==0.11.0
- torchvision ==0.12.0
- datasets *
- fastai *
- ipywidgets *
- jupyterlab *
- transformers *
- nvidia/cuda 12.1.0-base-ubuntu22.04
- nvidia/cuda 11.6.2-devel-ubuntu20.04 build
- nvidia/cuda 12.1.0-devel-ubuntu22.04 build
- suvash/deeplibs py3.11-cuda12.1-ubuntu22.04-<CORRECT-SHA> build
- torch ==2.1.2
- torchaudio ==2.1.2
- torchvision ==0.16.2
- nvidia/cuda 12.6.3-devel-ubuntu24.04 build
- torch ==2.6.0
- torchaudio ==2.6.0
- torchvision ==0.21.0
- python 3.11.11.*