Docker Example with PyTorch

Install

• https://docs.docker.com/engine/install/
• https://docs.docker.com/engine/install/linux-postinstall/
• docker run hello-world

Install NVIDIA Container Toolkit

• https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html
• sudo nvidia-ctk runtime configure --runtime=docker
• sudo systemctl restart docker

(Optional - this is how you would do it if you weren't using a custom image)

• docker pull pytorch/pytorch:2.4.1-cuda11.8-cudnn9-devel

Required - build and run custom image derived from official

• ./docker-build.sh
• ./docker-run.sh
• python3 ./pytorch_example.py