Skip to content

Commit 20b6c8d

Browse files
deep1712DEVAID-MSFT
deep1712
and
DEVAID-MSFT
authored
Fixing Nvidia Drivers to latest (#312)
* Adding -server * Adding Nvidia Drivers server * Adding metadata * Fixing comment --------- Co-authored-by: Deepanshu Vaid <[email protected]>
1 parent ed0ff3a commit 20b6c8d

File tree

6 files changed

+21
-16
lines changed

6 files changed

+21
-16
lines changed

‎src/VirtualClient/VirtualClient.Dependencies/CUDAAndNvidiaGPUDriverInstallation.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -343,7 +343,7 @@ private List<string> VersionSpecificInstallationCommands(LinuxDistribution linux
343343
case LinuxDistribution.Ubuntu:
344344
commands.Add("apt update");
345345
commands.Add("apt upgrade -y");
346-
commands.Add($"apt install nvidia-driver-{this.LinuxDriverVersion} nvidia-dkms-{this.LinuxDriverVersion} -y");
346+
commands.Add($"apt install nvidia-driver-{this.LinuxDriverVersion}-server nvidia-dkms-{this.LinuxDriverVersion}-server -y");
347347
commands.Add($"apt install cuda-drivers-fabricmanager-{this.LinuxDriverVersion} -y");
348348

349349
break;

‎src/VirtualClient/VirtualClient.Main/profiles/PERF-GPU-MLPERF-NVIDIA.json

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,16 @@
44
"RecommendedMinimumExecutionTime": "10:00:00",
55
"SupportedPlatforms": "linux-x64",
66
"SupportedOperatingSystems": "Ubuntu",
7+
"RecommendOperationSystem": "Ubunutu-22.04, Default LinusCudaVersion,LinuxDriverVersion,LinuxLocalRunFile support this OS only.",
78
"SupportedRequireCustomSystemSupport": "A100_PCIe_80GBx4",
89
"SpecialRequirements": "This is an Nvidia GPU-specialized workload. It depends upon the system having an Nvidia GPU card/chip."
910
},
1011
"Parameters": {
1112
"Username": "",
1213
"DiskFilter": "SizeGreaterThan:1000gb",
13-
"LinuxCudaVersion": "12.0",
14-
"LinuxDriverVersion": "525",
15-
"LinuxLocalRunFile": "https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda_12.0.0_525.60.13_linux.run",
14+
"LinuxCudaVersion": "12.4",
15+
"LinuxDriverVersion": "550",
16+
"LinuxLocalRunFile": "https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_550.54.15_linux.run",
1617
"RequireCustomSystemSupport": true
1718
},
1819
"Actions": [

‎src/VirtualClient/VirtualClient.Main/profiles/PERF-GPU-MLPERF-TRAINING-NVIDIA.json

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,16 +4,17 @@
44
"RecommendedMinimumExecutionTime": "10:00:00",
55
"SupportedPlatforms": "linux-x64",
66
"SupportedOperatingSystems": "Ubuntu",
7+
"RecommendOperationSystem": "Ubunutu-22.04, Default LinusCudaVersion,LinuxDriverVersion,LinuxLocalRunFile support this OS only.",
78
"SpecialRequirements": "This is an Nvidia GPU-specialized workload. It depends upon the system having an Nvidia GPU card/chip. Requires an OSDisk of >=256GB size and Data disk of >=8TB size",
89
"SupportedImplementations": "Visit the link https://github.com/mlcommons/training_results_v2.1/tree/main/NVIDIA/benchmarks enter the implementation folder to see supported implementations.",
910
"SupportedConfigFiles": "Visit the link https://github.com/mlcommons/training_results_v2.1/tree/main/NVIDIA/benchmarks/bert/implementations/pytorch-22.09 for each benchmark and implementation you can the supported "
1011
},
1112
"Parameters": {
1213
"Username": null,
1314
"DiskFilter": "BiggestSize",
14-
"LinuxCudaVersion": "12.0",
15-
"LinuxDriverVersion": "525",
16-
"LinuxLocalRunFile": "https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda_12.0.0_525.60.13_linux.run"
15+
"LinuxCudaVersion": "12.4",
16+
"LinuxDriverVersion": "550",
17+
"LinuxLocalRunFile": "https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_550.54.15_linux.run"
1718
},
1819
"Actions": [
1920
{

‎src/VirtualClient/VirtualClient.Main/profiles/PERF-GPU-MLPERF.json

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,16 @@
44
"RecommendedMinimumExecutionTime": "10:00:00",
55
"SupportedPlatforms": "linux-x64",
66
"SupportedOperatingSystems": "Ubuntu",
7+
"RecommendOperationSystem": "Ubunutu-22.04, Default drivers support this OS only.",
78
"SupportedRequireCustomSystemSupport": "A100_PCIe_80GBx4",
89
"SpecialRequirements": "This is an Nvidia GPU-specialized workload. It depends upon the system having an Nvidia GPU card/chip."
910
},
1011
"Parameters": {
1112
"Username": "",
1213
"DiskFilter": "SizeGreaterThan:1000gb",
13-
"LinuxCudaVersion": "12.0",
14-
"LinuxDriverVersion": "525",
15-
"LinuxLocalRunFile": "https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda_12.0.0_525.60.13_linux.run",
14+
"LinuxCudaVersion": "12.4",
15+
"LinuxDriverVersion": "550",
16+
"LinuxLocalRunFile": "https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_550.54.15_linux.run",
1617
"RequireCustomSystemSupport": true
1718
},
1819
"Actions": [

‎src/VirtualClient/VirtualClient.Main/profiles/PERF-GPU-SUPERBENCH-NVIDIA.json

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,14 +5,15 @@
55
"RecommendedMinimumExecutionTime": "08:00:00",
66
"SupportedPlatforms": "linux-x64",
77
"SupportedOperatingSystems": "Ubuntu",
8+
"RecommendOperationSystem": "Ubunutu-22.04, Default LinusCudaVersion,LinuxDriverVersion,LinuxLocalRunFile support this OS only.",
89
"SpecialRequirements": "This is an Nvidia GPU-specialized workload. It depends upon the system having an Nvidia GPU card/chip."
910
},
1011
"Parameters": {
1112
"ConfigurationFile": "default.yaml",
1213
"Username": "",
13-
"LinuxCudaVersion": "12.0",
14-
"LinuxDriverVersion": "525",
15-
"LinuxLocalRunFile": "https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda_12.0.0_525.60.13_linux.run"
14+
"LinuxCudaVersion": "12.4",
15+
"LinuxDriverVersion": "550",
16+
"LinuxLocalRunFile": "https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_550.54.15_linux.run"
1617
},
1718
"Actions": [
1819
{

‎src/VirtualClient/VirtualClient.Main/profiles/PERF-GPU-SUPERBENCH.json

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,14 +5,15 @@
55
"RecommendedMinimumExecutionTime": "08:00:00",
66
"SupportedPlatforms": "linux-x64",
77
"SupportedOperatingSystems": "Ubuntu",
8+
"RecommendOperationSystem": "Ubunutu-22.04, Default LinuxCudaVersion,LinuxDriverVersion,LinuxLocalRunFile support this OS only.",
89
"SpecialRequirements": "This is an Nvidia GPU-specialized workload. It depends upon the system having an Nvidia GPU card/chip."
910
},
1011
"Parameters": {
1112
"ConfigurationFile": "default.yaml",
1213
"Username": "",
13-
"LinuxCudaVersion": "12.0",
14-
"LinuxDriverVersion": "525",
15-
"LinuxLocalRunFile": "https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda_12.0.0_525.60.13_linux.run"
14+
"LinuxCudaVersion": "12.4",
15+
"LinuxDriverVersion": "550",
16+
"LinuxLocalRunFile": "https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_550.54.15_linux.run"
1617
},
1718
"Actions": [
1819
{

0 commit comments

Comments
 (0)