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