Docker Hub
https://registry.hub.docker.com/r/nvidia/cuda#!CUDA is a parallel computing platform and programming model developed by NVIDIA for general computing on graphical processing units (GPUs). With CUDA, developers can dramatically speed up computing applications by harnessing the power of GPUs. The CUDA Toolkit from NVIDIA provides everything you need to develop GPU-accelerated applications.
PyTorch Docker image - hub.docker.com
https://hub.docker.com/r/anibali/pytorch/#!You will also need to install the NVIDIA Container Toolkit to enable GPU device access within Docker containers. This can be found at NVIDIA/nvidia-docker. Prebuilt images Prebuilt images are available on Docker Hub under the name anibali/pytorch. For example, you can pull an image with PyTorch 1.8.1 and CUDA 11.1 using: