OS prerequisite changes (dell#2)

Add OS prerequisite for Gaudi drivers. Only ubuntu support --------- Signed-off-by: Yuhao Luo <[email protected]>
dweineha · Jul 18, 2024 · de7af57 · de7af57
1 parent 10651f8
commit de7af57
Show file tree

Hide file tree

Showing 4 changed files with 68 additions and 3 deletions.
diff --git a/input/config/ubuntu/22.04/gaudi.json b/input/config/ubuntu/22.04/gaudi.json
@@ -0,0 +1,51 @@
+{
+    "gaudi": {
+      "cluster": [
+      { 
+        "package": "habanalabs-dkms={{ gaudi_version }}",
+        "type": "deb",
+        "repo_name": "gaudi"
+      },
+      { 
+        "package": "habanalabs-firmware={{ gaudi_version }}",
+        "type": "deb",
+        "repo_name": "gaudi"
+      },
+      { 
+        "package": "habanalabs-firmware-tools={{ gaudi_version }}",
+        "type": "deb",
+        "repo_name": "gaudi"
+      },
+      { 
+        "package": "habanalabs-graph={{ gaudi_version }}",
+        "type": "deb",
+        "repo_name": "gaudi"
+      },
+      { 
+        "package": "habanalabs-qual={{ gaudi_version }}",
+        "type": "deb",
+        "repo_name": "gaudi"
+      },
+      { 
+        "package": "habanalabs-rdma-core={{ gaudi_version }}",
+        "type": "deb",
+        "repo_name": "gaudi"
+      },
+      { 
+        "package": "habanalabs-thunk={{ gaudi_version }}",
+        "type": "deb",
+        "repo_name": "gaudi"
+      },
+      { 
+        "package": "habanatools={{ gaudi_version }}",
+        "type": "deb",
+        "repo_name": "gaudi"
+      },
+      { 
+        "package": "habanalabs-container-runtime={{ gaudi_version }}",
+        "type": "deb",
+        "repo_name": "gaudi"
+      }
+      ]
+    }
+  }
diff --git a/input/config/ubuntu/22.04/pytorch.json b/input/config/ubuntu/22.04/pytorch.json
@@ -41,6 +41,17 @@
       }
     ]
 
+  },
+  "pytorch_gaudi": { 
+
+    "cluster": [ 	
+      { 
+        "package": "vault.habana.ai/gaudi-docker/1.16.2/ubuntu22.04/habanalabs/pytorch-installer-2.2.2",
+        "tag": "latest",
+        "type": "image"
+      }
+    ]
+
   }
 
-}
+}
diff --git a/input/local_repo_config.yml b/input/local_repo_config.yml
@@ -134,3 +134,4 @@ omnia_repo_url_ubuntu:
   - { url: "https://nvidia.github.io/libnvidia-container/stable/deb/amd64 /", gpgkey: "https://nvidia.github.io/libnvidia-container/gpgkey" }
   - { url: "http://ppa.launchpad.net/deadsnakes/ppa/ubuntu {{ os_release }} main", gpgkey: "" }
   - { url: "https://a2o.github.io/snoopy-packages/repo/ubuntu {{ os_release }} stable", publickey: "https://a2o.github.io/snoopy-packages/snoopy-packages-key.pub" }
+  - { url: "https://vault.habana.ai/artifactory/debian {{ os_release }} main", publickey: " https://vault.habana.ai/artifactory/api/gpg/key/public" }
diff --git a/input/software_config.json b/input/software_config.json
@@ -11,7 +11,8 @@
         {"name": "roce_plugin"},
         {"name": "jupyter"},
         {"name": "pytorch"},
-        {"name": "tensorflow"}
+        {"name": "tensorflow"},
+        {"name": "gaudi", "version": "1.16.2-2"}
     ],
 
     "bcm_roce": [
@@ -23,7 +24,8 @@
     "pytorch": [
         {"name": "pytorch_cpu"},
         {"name": "pytorch_amd"},
-        {"name": "pytorch_nvidia"}
+        {"name": "pytorch_nvidia"},
+        {"name": "pytorch_gaudi"}
     ],
     "tensorflow": [
         {"name": "tensorflow_cpu"},