diff --git a/scripts/setup-centos9.sh b/scripts/setup-centos9.sh index f381bf5d1b64..04224adc90d1 100755 --- a/scripts/setup-centos9.sh +++ b/scripts/setup-centos9.sh @@ -221,7 +221,7 @@ function install_cuda { # See https://developer.nvidia.com/cuda-downloads dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel9/x86_64/cuda-rhel9.repo local dashed="$(echo $1 | tr '.' '-')" - dnf install -y cuda-nvcc-$dashed cuda-cudart-devel-$dashed cuda-nvrtc-devel-$dashed cuda-driver-devel-$dashedcuda-driver-devel-$dashed cuda-driver-devel-$dashed + dnf install -y cuda-nvcc-$dashed cuda-cudart-devel-$dashed cuda-nvrtc-devel-$dashed cuda-driver-devel-$dashed } function install_velox_deps { diff --git a/scripts/setup-ubuntu.sh b/scripts/setup-ubuntu.sh index fbaf63afc442..5874e98e9651 100755 --- a/scripts/setup-ubuntu.sh +++ b/scripts/setup-ubuntu.sh @@ -245,7 +245,7 @@ function install_cuda { $SUDO apt update fi local dashed="$(echo $1 | tr '.' '-')" - $SUDO apt install -y cuda-nvcc-$dashed cuda-cudart-dev-$dashed cuda-nvrtc-dev-$dashed cuda-driver-devel-$dashed + $SUDO apt install -y cuda-nvcc-$dashed cuda-cudart-dev-$dashed cuda-nvrtc-dev-$dashed cuda-driver-dev-$dashed } function install_velox_deps {