chore: install NVIDIA docker

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

.circleci/config.yml

@@ -174,6 +174,22 @@ commands:
           name: Dump apt sources
           command: cat /etc/apt/sources.list
 
+  install-nvidia-docker:
+    description: "Install NVIDIA Docker"
+    steps:
+      - run:
+          name: Install NVIDIA Docker
+          command: |
+            distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
+            && curl -fsSL 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://#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 nvidia-ctk runtime configure --runtime=docker
+            sudo systemctl restart docker
+
   create-env:
     description: "Install dependencies for Torch-TensorRT"
     parameters:
@@ -1075,6 +1091,7 @@ jobs:
           condition: << parameters.enabled >>
           steps:
             - checkout
+            - install-nvidia-docker
             - run:
                 name: "Build packaging container"
                 command: |