A step-by-step guide to setting up Nvidia GPUs with CUDA support running on Docker (and Compose) containers on NixOS host
python deep-learning docker-compose nixos tensorflow pytorch nvidia triton nvidia-smi nvidia-docker nvidia-cuda jax
-
Updated
Jul 17, 2024 - Dockerfile