Skip to content
This repository has been archived by the owner on Jan 12, 2023. It is now read-only.

Commit

Permalink
Fix logic for driver URL calculation
Browse files Browse the repository at this point in the history
Found all the drivers availble in the google storage buckets using:
`gsutil ls -r gs://nvidia-drivers-us-public/ | grep -E -i -v
'sha256|.gcmanifest|GRID' | grep ".run" | sort | uniq -u`

The driver URL logic method has been retro fitted to retrieve all
versions availble in the storage bucket.
  • Loading branch information
arnav-kansal authored and rkolchmeyer committed Feb 4, 2021
1 parent 5e39ee4 commit db31e9f
Showing 1 changed file with 7 additions and 6 deletions.
13 changes: 7 additions & 6 deletions cos-gpu-installer-docker/gpu_installer_url_lib.sh
Original file line number Diff line number Diff line change
Expand Up @@ -53,14 +53,15 @@ default_installer_download_url() {
if (( "${major_version}" < 390 )); then
# Versions prior to 390 are downloaded from the upstream location.
echo "https://us.download.nvidia.com/tesla/${driver_version}/NVIDIA-Linux-x86_64-${driver_version}.run"
elif (( "${major_version}" == 390 )); then
# The naming format changed after version 390.
elif (( "${major_version}" == 390 )) && (( "${minor_version}" == 46 )); then
# 390.46 is the only version residing in the TESLSA/ dir
echo "https://storage.googleapis.com/nvidia-drivers-${download_location}-public/TESLA/NVIDIA-Linux-x86_64-${driver_version}.run"
elif (( "${major_version}" >= 396 )) && (( "${minor_version}" >= 37 )); then
# Apparently the naming format changed again starting since 396.37.
echo "https://storage.googleapis.com/nvidia-drivers-${download_location}-public/tesla/${driver_version}/NVIDIA-Linux-x86_64-${driver_version}.run"
else
elif (( "${major_version}" == 396 )) && (( "${minor_version}" == 26 )); then
# Different naming format for 396.26 including the -dignostic keyword.
echo "https://storage.googleapis.com/nvidia-drivers-${download_location}-public/tesla/${driver_version}/NVIDIA-Linux-x86_64-${driver_version}-diagnostic.run"
else
# All other versions available in the gs conform to this naming convention.
echo "https://storage.googleapis.com/nvidia-drivers-${download_location}-public/tesla/${driver_version}/NVIDIA-Linux-x86_64-${driver_version}.run"
fi
}

Expand Down

0 comments on commit db31e9f

Please sign in to comment.