use cuda11.8 for building wheels

benfred · Nov 21, 2023 · 5483e56 · 5483e56
1 parent 1711f6a
commit 5483e56
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/ci/install_cuda.sh b/ci/install_cuda.sh
@@ -1,9 +1,9 @@
 yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-rhel7.repo
 yum install --setopt=obsoletes=0 -y \
-    cuda-nvcc-11-2 \
-    cuda-cudart-devel-11-2 \
-    libcurand-devel-11-2 \
-    libcublas-devel-11-2 \
-    cuda-nvprof-11-2 \
+    cuda-nvcc-11-8 \
+    cuda-cudart-devel-11-8 \
+    libcurand-devel-11-8 \
+    libcublas-devel-11-8 \
+    cuda-nvprof-11-8 \
     ninja-build
-ln -s cuda-11.2 /usr/local/cuda
+ln -s cuda-11.8 /usr/local/cuda