Oya-Tomo / docker-pytorch-nvidia

docker-pytorch-nvidia compose template

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