update setup script

Author: blooop

scripts/setup_host.sh

@@ -10,7 +10,7 @@ for pkg in docker.io docker-doc docker-compose docker-compose-v2 podman-docker c
 
 # Add Docker's official GPG key:
 sudo apt-get update
-sudo apt-get install ca-certificates curl
+sudo apt-get install -y ca-certificates curl
 sudo install -m 0755 -d /etc/apt/keyrings
 sudo curl -fsSL https://download.docker.com/linux/ubuntu/gpg -o /etc/apt/keyrings/docker.asc
 sudo chmod a+r /etc/apt/keyrings/docker.asc
@@ -23,7 +23,7 @@ echo \
 sudo apt-get update
 
 # Install docker
-sudo apt-get install docker-ce docker-ce-cli containerd.io docker-buildx-plugin docker-compose-plugin
+sudo apt-get install -y docker-ce docker-ce-cli containerd.io docker-buildx-plugin docker-compose-plugin
 #END OFFICIAL DOCKER INSTALL
 
 sudo groupadd docker
@@ -37,11 +37,11 @@ curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dear
     sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
     sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
 sudo apt-get update
-sudo apt-get install -y nvidia-container-toolkit
+sudo apt-get install -y nvidia-docker2 
 sudo nvidia-ctk runtime configure --runtime=docker
 sudo systemctl restart docker
 
-sudo apt install git-lfs
+sudo apt install -y git-lfs
 
 #Install rocker and rocker extensions which are used to launch the devcontainer
 # pip install rocker off-your-rocker git+https://github.com/blooop/deps_rocker