Fix typo

facebookresearch · Nov 18, 2024 · de81c6d · de81c6d
1 parent 59bd75c
commit de81c6d
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/.github/actions/install_ubuntu_gpu_deps/action.yml b/.github/actions/install_ubuntu_gpu_deps/action.yml
@@ -8,9 +8,9 @@ runs:
     shell: bash
     run: |
       NVIDIA_SMI_OUTPUT=$(nvidia-smi)
-      DRIVER_VERSION=$(echo "$NVIDIA_SMI_OUTPUT" | grep -oP 'Driver Version:\s+\K[\d.]+')
-      MAJOR_VERSION=$(echo "$NVIDIA_SMI_OUTPUT" | grep -oP 'Driver Version:\s+\K\d+')
-      UBUNTU_PACKAGE_SUFFIX="${MAJOR_VERSION}=${DRIVER_VERSION}-0ubuntu1"
+      FULL_DRIVER_VERSION=$(echo "$NVIDIA_SMI_OUTPUT" | grep -oP 'Driver Version:\s+\K[\d.]+')
+      MAJOR_DRIVER_VERSION=$(echo "$NVIDIA_SMI_OUTPUT" | grep -oP 'Driver Version:\s+\K\d+')
+      UBUNTU_PACKAGE_SUFFIX="${MAJOR_DRIVER_VERSION}=${FULL_DRIVER_VERSION}-0ubuntu1"
       sudo apt-get update
-      sudo apt-get install -y libnvidia-common-${DRIVER_VERSION}
-      sudo apt-get install -y libnvidia-gl-${DRIVER_VERSION}
+      sudo apt-get install -y libnvidia-common-${UBUNTU_PACKAGE_SUFFIX}
+      sudo apt-get install -y libnvidia-gl-${UBUNTU_PACKAGE_SUFFIX}