docker.io/nvcr.io/nvidia/pytorch:24.01-py3
docker.io/nvcr.io/nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04