Update build_wheel_cuda.yml

sandialabs · Aug 21, 2024 · e2427d9 · e2427d9
1 parent bf449a6
commit e2427d9
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/.github/workflows/build_wheel_cuda.yml b/.github/workflows/build_wheel_cuda.yml
@@ -27,8 +27,7 @@ jobs:
         env:
           CIBW_BUILD: cp38-manylinux_x86_64
           CIBW_BUILD_VERBOSITY: 1
-          #CIBW_BEFORE_ALL_LINUX: "yum install -y openblas-devel nvidia-cuda-toolkit"
-          CIBW_BEFORE_ALL_LINUX: "apt install -y nvidia-cuda-toolkit"
+          CIBW_BEFORE_ALL_LINUX: "wget https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-repo-rhel7-10.2.89-1.x86_64.rpm && rpm -i cuda-repo-*.rpm && yum install -y openblas-devel cuda"
           CIBW_CONFIG_SETTINGS: "cmake.define.ENABLE_PYTHON_EMBED=OFF cmake.define.PYGENTEN_CUDA=ON"
 
       - name: Upload distributions