Adding DCGMI linux-arm64 support (#377)

* Adding DCGMI arm64 support

* Adding DCGMI installation for linux-arm64

* Adding ppc64le link

* Adding username

---------

Co-authored-by: Deepanshu Vaid <devaid@microsoft.com>
Co-authored-by: nmalkapuram <nmalkapuram@microsoft.com>

Author: 3 people

src/VirtualClient/VirtualClient.Actions/DCGMI/DCGMIExecutor.cs

@@ -122,6 +122,17 @@ public string ListOfFieldIDsDmon
             }
         }
 
+        /// <summary>
+        /// The version of CUDA to be installed in Linux Systems
+        /// </summary>
+        public string LinuxCudaVersion
+        {
+            get
+            {
+                return this.Parameters.GetValue<string>(nameof(DCGMIExecutor.LinuxCudaVersion), "11.6");
+            }
+        }
+
         /// <summary>
         /// Initializes enviroment to run DCGMI.
         /// </summary>
@@ -361,7 +372,7 @@ private async Task ExecuteDCGMIProfTesterSubsystemAsync(EventContext telemetryCo
             using (BackgroundOperations profiling = BackgroundOperations.BeginProfiling(this, cancellationToken))
             {
                 List<Task<string>> tasksList = new List<Task<string>>();
-                string dcgmproftestercommand = $"/usr/bin/dcgmproftester11 --no-dcgm-validation -t {this.FieldIDProftester} -d 10";
+                string dcgmproftestercommand = $"/usr/bin/dcgmproftester{(int)Convert.ToDouble(this.LinuxCudaVersion)} --no-dcgm-validation -t {this.FieldIDProftester} -d 10";
                 string dmoncommand = $"dcgmi dmon -e {this.ListOfFieldIDsDmon} -c 15";
                 DateTime startTime = DateTime.UtcNow;
 

src/VirtualClient/VirtualClient.Dependencies/DCGMIInstallation.cs

@@ -107,15 +107,19 @@ private async Task InstallDCGMIUbuntuOrDebianAsync(EventContext telemetryContext
             List<string> commands = new List<string>();
 
             string delKeyCommand = "apt-key del 7fa2af80";
-            string metaPackageDownloadCommand;
+            string metaPackageDownloadCommand = string.Empty;
             if (this.PlatformSpecifics.CpuArchitecture == Architecture.X64)
             {
                 metaPackageDownloadCommand = $@"bash -c ""wget https://developer.download.nvidia.com/compute/cuda/repos/$(echo $(. /etc/os-release; echo $ID$VERSION_ID) | sed -e 's/\.//g')/x86_64/cuda-keyring_1.0-1_all.deb""";
             }
-            else
+            else if (this.PlatformSpecifics.CpuArchitecture == Architecture.Arm64) 
             {
                 metaPackageDownloadCommand = $@"bash -c ""wget https://developer.download.nvidia.com/compute/cuda/repos/$(echo $(. /etc/os-release; echo $ID$VERSION_ID) | sed -e 's/\.//g')/sbsa/cuda-keyring_1.0-1_all.deb""";
             }
+            else if (this.PlatformSpecifics.CpuArchitecture == Architecture.Ppc64le)
+            {
+                metaPackageDownloadCommand = $@"bash -c ""wget https://developer.download.nvidia.com/compute/cuda/repos/$(echo $(. /etc/os-release; echo $ID$VERSION_ID) | sed -e 's/\.//g')/ppc64le/cuda-keyring_1.0-1_all.deb""";
+            }
 
             string cudaGPGKeyInstallCommand = @"dpkg -i cuda-keyring_1.0-1_all.deb";
             string updateCommand = @"apt-get update";

src/VirtualClient/VirtualClient.Main/profiles/QUAL-GPU-DCGMI.json

@@ -7,7 +7,8 @@
     },
     "Parameters": {
         "Level": "4",
-        "Username": "null"
+        "Username": "",
+        "LinuxCudaVersion": "12.4"
     },
     "Actions": [
         {
@@ -59,6 +60,7 @@
                 "Scenario": "CUDAIntegrationDiagnostics",
                 "Subsystem": "CUDATestGenerator",
                 "FieldIDProftester": "1004",
+                "LinuxCudaVersion": "$.Parameters.LinuxCudaVersion",
                 "ListOfFieldIDsDmon": "1004,1005,1001"
             }
         }
@@ -68,7 +70,7 @@
             "Type": "NvidiaCudaInstallation",
             "Parameters": {
                 "Scenario": "InstallNvidiaCuda",
-                "LinuxCudaVersion": "11.6",
+                "LinuxCudaVersion": "$.Parameters.LinuxCudaVersion",
                 "LinuxDriverVersion": "510",
                 "Username": "$.Parameters.Username",
                 "LinuxLocalRunFile": "https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux.run"