apt install cuda

 RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
 RUN sudo dpkg -i cuda-keyring_1.1-1_all.deb
 RUN sudo apt update --fix-missing
 RUN sudo apt install -y cuda

See also

• GPU, CUDA, ROCm, Stable Diffusion, cuda-keyring, nvcc, NVIDIA Data Center GPU Manager (DCGM)