nvidia docker doesn't have nvcc installed?

[YuxinxinChen]

Hi I am using nvidia docker but found there is no nvcc installed inside the container. Here is what I did:

docker pull nvcr.io/nvidia/cuda:12.0.1-base-ubuntu22.04
$ docker run -it --rm nvcr.io/nvidia/cuda:12.0.1-base-ubuntu22.04 /bin/bash
root@ccc9afff5b92:/# nvcc --version
bash: nvcc: command not found

I am doing this on a slurm login node which has no GPU, but the login node has cuda installed though with different version. But I suppose the benefit of a docker container is to allow us use different cuda version easily.

Could you tell me where I did wrong?

Thanks!

[elezar]

For nvcc you need the nvcr.io/nvidia/cuda:12.0.1-devel-ubuntu22.04 container.

[YuxinxinChen]

Thanks!

[iphysresearch]

FROM nvidia/cuda12.3.2-cudnn9-devel-ubuntu22.04 works for me!