Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

chore: bumping the grid driver affecting linux kernels 5.15.1063+ versions #381

Merged
merged 6 commits into from
Jun 4, 2024
Merged

chore: bumping the grid driver affecting linux kernels 5.15.1063+ versions #381

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
0883c70
chore: bumping the grid driver affecting linux kernels 5.15.1063+ ver…
Bryce-Soghigian May 31, 2024
fc40b9e
chore: bumping the cuda driver versions too
Bryce-Soghigian May 31, 2024
d3d43d1
chore: fix gpu sha
Bryce-Soghigian May 31, 2024
b92539f
fix: use different driver version for 550
Bryce-Soghigian Jun 1, 2024
aac0c46
fix: bumping major version to 550
Bryce-Soghigian Jun 3, 2024
d601e88
ci
Bryce-Soghigian Jun 3, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 7 additions & 5 deletions pkg/utils/gpu.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,11 +23,13 @@ import (
// TODO: Get these from agentbaker
const (
Nvidia470CudaDriverVersion = "cuda-470.82.01"
Nvidia535CudaDriverVersion = "cuda-535.54.03"
Nvidia535GridDriverVersion = "grid-535.54.03"
Nvidia550CudaDriverVersion = "cuda-550.54.15"
Nvidia535GridDriverVersion = "grid-535.161.08"

AKSGPUGridSHA = "sha-20ffa2"
AKSGPUCudaSHA = "sha-ff213d"
// These SHAs will change once we update aks-gpu images in aks-gpu repository. We do that fairly rarely at this time.
// So for now these will be kept here like this and periodically bump them
AKSGPUGridSHA = "sha-d1f0ca"
AKSGPUCudaSHA = "sha-2d4c96"
)

func GetAKSGPUImageSHA(size string) string {
Expand Down Expand Up @@ -146,7 +148,7 @@ func GetGPUDriverVersion(size string) string {
if isStandardNCv1(size) {
return Nvidia470CudaDriverVersion
}
return Nvidia535CudaDriverVersion
return Nvidia550CudaDriverVersion
}

func isStandardNCv1(size string) bool {
Expand Down
6 changes: 3 additions & 3 deletions pkg/utils/gpu_test.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -55,9 +55,9 @@ func TestGetGPUDriverVersion(t *testing.T) {
}{
{"GRID Driver - NV Series v5", "standard_nv6ads_a10_v5", Nvidia535GridDriverVersion},
{"CUDA Driver - NC Series v1", "standard_nc6s", Nvidia470CudaDriverVersion},
{"CUDA Driver - NC Series v2", "standard_nc6s_v2", Nvidia535CudaDriverVersion},
{"Unknown SKU", "unknown_sku", Nvidia535CudaDriverVersion},
{"CUDA Driver - NC Series v3", "standard_nc6s_v3", Nvidia535CudaDriverVersion},
{"CUDA Driver - NC Series v2", "standard_nc6s_v2", Nvidia550CudaDriverVersion},
{"Unknown SKU", "unknown_sku", Nvidia550CudaDriverVersion},
{"CUDA Driver - NC Series v3", "standard_nc6s_v3", Nvidia550CudaDriverVersion},
{"GRID Driver - A10", "standard_nc8ads_a10_v4", Nvidia535GridDriverVersion},
}

Expand Down