Missing function in docker build

If I build the docker using the provided `docker/Dockerfile`, with the following two lines

```
ARG CUDA=11.8.0
FROM nvidia/cuda:${CUDA}-cudnn8-runtime-ubuntu22.04
```
When I run `docker/run_docker.py`, I get the error

`I0125 13:11:56.513862 140328062711616 run_docker.py:258] jaxlib.xla_extension.XlaRuntimeError: INTERNAL: Could not find the corresponding function`

But if I build using the cuda with the `devel` label by changing the second line of the above to

```
FROM nvidia/cuda:${CUDA}-cudnn8-devel-ubuntu22.04
```

The docker runs fine. So something is missing in the linked library when building using cuda with the `runtime` label.