machine learnign notebook on edge gpu 101

Azure-Samples · Sep 2, 2020 · 7d6eec1 · 7d6eec1
1 parent 389380e
commit 7d6eec1
Show file tree

Hide file tree

Showing 5 changed files with 1,236 additions and 0 deletions.
diff --git a/machine-learning-notebooks/gpu-accleration-onedge/img/conf_file_download.JPG b/machine-learning-notebooks/gpu-accleration-onedge/img/conf_file_download.JPG
diff --git a/machine-learning-notebooks/gpu-accleration-onedge/img/ml-flow-edge.png b/machine-learning-notebooks/gpu-accleration-onedge/img/ml-flow-edge.png
diff --git a/machine-learning-notebooks/gpu-accleration-onedge/iotedge-tf-template-gpu.json b/machine-learning-notebooks/gpu-accleration-onedge/iotedge-tf-template-gpu.json
@@ -0,0 +1,157 @@
+{
+
+  "modulesContent": {
+
+    "$edgeAgent": {
+
+      "properties.desired": {
+
+        "schemaVersion": "1.0",
+
+        "runtime": {
+
+          "type": "docker",
+
+          "settings": {
+
+            "minDockerVersion": "v1.25",
+
+            "loggingOptions": "",
+
+            "registryCredentials": {
+
+              "__REGISTRY_NAME": {
+
+          			"username": "__REGISTRY_USER_NAME",
+
+          			"password":"__REGISTRY_PASSWORD",
+
+          			"address":"__REGISTRY_NAME.azurecr.io"
+
+          		}
+
+            }
+
+          }
+
+        },
+
+        "systemModules": {
+
+          "edgeAgent": {
+
+            "type": "docker",
+
+            "settings": {
+
+              "image": "mcr.microsoft.com/azureiotedge-agent:1.0",
+
+              "createOptions": "{}",
+
+              "env": {
+
+                "UpstreamProtocol": {
+
+                  "value": "MQTT"
+
+                }
+
+              }
+
+            }
+
+          },
+
+          "edgeHub": {
+
+            "type": "docker",
+
+            "status": "running",
+
+            "restartPolicy": "always",
+
+            "settings": {
+
+              "image": "mcr.microsoft.com/azureiotedge-hub:1.0",
+
+              "createOptions": "{\"User\":\"root\",\"HostConfig\":{\"PortBindings\":{\"5671/tcp\":[{\"HostPort\":\"5671\"}], \"8883/tcp\":[{\"HostPort\":\"8883\"}],\"443/tcp\":[{\"HostPort\":\"443\"}]}}}",
+
+              "env": {
+
+                "UpstreamProtocol": {
+
+                  "value": "MQTT "
+
+                }
+
+              }
+
+            }
+
+          }
+
+        },
+
+        "modules": {
+
+
+          "__MODULE_NAME": {
+
+            "version": "1.0",
+
+            "type": "docker",
+
+            "status": "running",
+
+            "restartPolicy": "always",
+
+            "settings": {
+
+              "image": "__REGISTRY_IMAGE_LOCATION",
+
+              "createOptions": "{\"HostConfig\":{\"PortBindings\":{\"5001/tcp\":[{\"HostPort\":\"5001\"}], \"8883/tcp\":[{\"HostPort\":\"5002\"}],},\"runtime\":\"nvidia\"},\"WorkingDir\":\"/var/azureml-app\"}"
+
+
+            }
+
+          }
+
+        }
+
+      }
+
+    },
+
+    "$edgeHub": {
+
+      "properties.desired": {
+
+        "schemaVersion": "1.0",
+
+        "routes": {
+
+
+          "machineLearningToIoTHub": "FROM /messages/modules/__MODULE_NAME/outputs/amlOutput INTO $upstream"
+
+        },
+
+        "storeAndForwardConfiguration": {
+
+          "timeToLiveSecs": 7200
+
+        }
+
+      }
+
+    },
+
+
+    "__MODULE_NAME": {
+
+      "properties.desired": {}
+
+    }
+
+  }
+
+}