Skip to content

Commit

Permalink
fix: nvidia version on el8
Browse files Browse the repository at this point in the history
  • Loading branch information
itkovian committed Dec 16, 2024
1 parent 9984765 commit 30a0b7d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ OUR_RELEASE=${RELEASE:-1}
# allow _empty_ version, which is used in pipeline

if grep "release 8.8" /etc/redhat-release; then
NVIDIA_DRIVER=${NVIDIA_DRIVER-550.90.07}
NVIDIA_DRIVER=${NVIDIA_DRIVER-555.42.06}
NVDRV_NVML_PKG="nvidia-driver-NVML${NVIDIA_DRIVER:+-$NVIDIA_DRIVER}"
CUDA_VERSION=${CUDA_VERSION:-12.6}
CUDA_NVML_PKG="cuda-nvml-devel-${CUDA_VERSION//./-}"
Expand Down

0 comments on commit 30a0b7d

Please sign in to comment.