-
docker run --name node-nvidia -ti --rm --gpus=all node:21.7.1-bookworm-slim bash
-
Check via
printenv
if the ENV-VarNVIDIA_VISIBLE_DEVICES=all
is set -
Run the install and cleanup command
sed -i 's/main/main contrib non-free non-free-firmware/g' /etc/apt/sources.list.d/debian.sources \
&& apt update \
&& apt install -y --no-install-recommends --no-install-suggests nvidia-driver \
&& apt-get autoremove -y --purge \
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
- Check if everything is installed via double tab
nvidia-smi
- Save the container as an image
docker commit node-nvidia jammsen/node-nvidia-smi-base:21.7.1-bookworm-slim