code refactor

facebookincubator · Feb 25, 2025 · 918f0ed · 918f0ed
1 parent 28c1f2d
commit 918f0ed
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/scripts/setup-centos9.sh b/scripts/setup-centos9.sh
@@ -80,7 +80,9 @@ function install_conda {
 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
-  dnf install -y cuda-nvcc-$(echo $1 | tr '.' '-') cuda-cudart-devel-$(echo $1 | tr '.' '-')
+  local dashed="$(echo $1 | tr '.' '-')"
+
+  dnf install -y cuda-nvcc-$dashed cuda-cudart-devel-$dashed cuda-nvrtc-devel-$dashed cuda-driver-devel-$dashed
 }
 
 function install_s3 {