Skip to content

Commit

Permalink
Updated README
Browse files Browse the repository at this point in the history
  • Loading branch information
djthorpe committed Aug 11, 2024
1 parent 164eb76 commit 23da71d
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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 <name> --rm \
--runtime nvidia --gpus all
<image> <arguments>
```

0 comments on commit 23da71d

Please sign in to comment.