nvidia-docker-container installation on Ubuntu

2. Install NVIDIA Container Toolkit

 

distribution=$(. /etc/os-release; echo $ID$VERSION_ID)

sudo apt update
sudo apt install -y curl gnupg lsb-release

curl -s -L https://nvidia.github.io/libnvidia-container/gpgkey \
  | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg

curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list \
  | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#' \
  | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list > /dev/null

sudo apt update
sudo apt install -y nvidia-container-toolkit

 

 

3. Configure Docker to use NVIDIA runtime

sudo nvidia-ctk runtime configure --runtime=docker
sudo systemctl restart docker

 

 

 

4. Test it

sudo docker run --gpus all -it --rm --ipc=host --shm-size=8g -v /home/dooseop/DATASET/ETRI:/workspace openpcdet-env:torch2.3.1 /bin/bash