From d49d5904363713ea8e45b2cc76704054ba6bdf25 Mon Sep 17 00:00:00 2001 From: "Nick J. Browning" Date: Mon, 2 Dec 2024 00:46:36 +0100 Subject: [PATCH] not sure if I need this. --- .github/workflows/main.yaml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/main.yaml b/.github/workflows/main.yaml index 0e6428a..530da90 100644 --- a/.github/workflows/main.yaml +++ b/.github/workflows/main.yaml @@ -71,9 +71,9 @@ jobs: CIBW_SKIP: "*-musllinux* *-win32 *-manylinux_i686" CIBW_ARCHS: "${{ matrix.cibw-arch }}" CIBW_MANYLINUX_X86_64_IMAGE: "manylinux2014_${{ matrix.cibw-arch }}" - #CIBW_ENVIRONMENT: > - # CUDA_HOME=/usr/local/cuda - # PIP_EXTRA_INDEX_URL=https://download.pytorch.org/whl/cu${{ matrix.cuda-version.replace('.', '') }} + CIBW_ENVIRONMENT: > + CUDA_HOME=/usr/local/cuda + PIP_EXTRA_INDEX_URL="https://download.pytorch.org/whl/cu${{ env.CUVERSION }}" CIBW_REPAIR_WHEEL_COMMAND_LINUX: | auditwheel repair --exclude libcuda.so --exclude libcuda.so.1 --exclude libc10.so --exclude libtorch.so --exclude libtorch_cpu.so --exclude libtorch_cuda.so --exclude libc10_cuda.so --exclude libcudart.so --exclude libnvToolsExt.so --exclude libnvrtc.so --exclude libnvrtc.so.12 -w {dest_dir} {wheel}