From 4f1d08785ffff87cef4104ee3052d2c2f1a2fc5c Mon Sep 17 00:00:00 2001 From: frostedoyster Date: Wed, 14 Feb 2024 13:10:25 +0100 Subject: [PATCH] Make it work --- .github/workflows/build-cuda-wheels.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/build-cuda-wheels.yml b/.github/workflows/build-cuda-wheels.yml index 8b98c6f50..5609076e8 100644 --- a/.github/workflows/build-cuda-wheels.yml +++ b/.github/workflows/build-cuda-wheels.yml @@ -71,7 +71,7 @@ jobs: CIBW_SKIP: "*-musllinux* *-win32 *-manylinux_i686" CIBW_MANYLINUX_X86_64_IMAGE: manylinux2014 # set environment variables for sphericart-torch build - CIBW_ENVIRONMENT: SPHERICART_ARCH_NATIVE=OFF CUDACXX=/usr/local/cuda/bin/nvcc TORCH_CUDA_ARCH_LIST=All CUDAARCHS=all SPHERICART_TORCH_TORCH_VERSION=${{ matrix.torch }} PIP_EXTRA_INDEX_URL=https://download.pytorch.org/whl/cu${{ cuda_version_without_decimal_point }} + CIBW_ENVIRONMENT: SPHERICART_ARCH_NATIVE=OFF CUDACXX=/usr/local/cuda/bin/nvcc TORCH_CUDA_ARCH_LIST=All CUDAARCHS=all SPHERICART_TORCH_TORCH_VERSION=${{ matrix.torch }} PIP_EXTRA_INDEX_URL=https://download.pytorch.org/whl/cu${{ env.cuda_version_without_decimal_point }} # do not complain for missing libtorch.so in sphericart-torch wheel CIBW_REPAIR_WHEEL_COMMAND_LINUX: | auditwheel repair --exclude libtorch.so --exclude libtorch_cpu.so --exclude libtorch_cuda.so --exclude libc10.so --exclude libc10_cuda.so -w {dest_dir} {wheel}