docker-pytorch-nvidia ML template prerequisites install docker engine https://docs.docker.com/engine/install/ install cuda / driver https://developer.nvidia.com/cuda-downloads install nvidia-container-toolkit https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html image tag pytorch/pytorch:2.3.1-cuda11.8-cudnn8-devel