From e2427d9107e43556fc05232c89ebfa083fd97ad2 Mon Sep 17 00:00:00 2001 From: Eric Phipps Date: Wed, 21 Aug 2024 08:24:13 -0600 Subject: [PATCH] Update build_wheel_cuda.yml --- .github/workflows/build_wheel_cuda.yml | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/.github/workflows/build_wheel_cuda.yml b/.github/workflows/build_wheel_cuda.yml index 7035d625c1..92a21e27f6 100644 --- 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