FIX install correct cupy in cuda 12 wheel tests

rapidsai · Mar 3, 2024 · ff29ec7 · ff29ec7
1 parent 9824dff
commit ff29ec7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh
@@ -42,6 +42,7 @@ sed -r -i "s/pylibraft(.*)\"/pylibraft${PACKAGE_CUDA_SUFFIX}\1${alpha_spec}\"/g"
 
 if [[ $PACKAGE_CUDA_SUFFIX == "-cu12" ]]; then
     sed -i "s/cuda-python[<=>\.,0-9a]*/cuda-python>=12.0,<13.0a0/g" ${pyproject_file}
+    sed -i "s/cupy-cuda11x/cupy-cuda12x/g" ${pyproject_file}
 fi
 
 cd "${package_dir}"