Repository which has some base images for running CUDA and cuDNN on Intel and ARM architectures.
If you want to use an NVIDIA GPU, You can use the following two images as the basis for your own images:
ghcr.io/mutablelogic/cuda-dev:1.0.2
- This image is based on Ubuntu 22.04 and includes the 12.6 CUDA toolkit and compiler build toolsghcr.io/mutablelogic/cuda-rt:1.0.2
- This image is based on Ubuntu 22.04 and includes the 12.6 CUDA runtime libraries.
When running a runtime container, then install the NVIDIA Container Toolkit first. Then you can run the container with the following command:
docker run \
--name <name> --rm \
--runtime nvidia --gpus all
<image> <arguments>