From 23da71dab5cccaa92c164dca549e41df08f5aab9 Mon Sep 17 00:00:00 2001 From: David Thorpe Date: Sun, 11 Aug 2024 10:36:43 +0200 Subject: [PATCH] Updated README --- README.md | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 58c9d06..cdb8cc1 100755 --- a/README.md +++ b/README.md @@ -4,7 +4,16 @@ Repository which has some base images for running CUDA and cuDNN on Intel and AR ## CUDA Images -If you want to use an NVIDIA GPU, then install the [NVIDIA Container Toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html) first. You can use the following two images as the basis for your own images: +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 tools * `ghcr.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](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html) first. Then you can run the container with the following command: + +```bash +docker run \ + --name --rm \ + --runtime nvidia --gpus all + +```