There was an error while loading. Please reload this page.
1 parent 0aee99a commit 56ff218Copy full SHA for 56ff218
1 file changed
.github/workflows/wheels.yml
@@ -56,9 +56,12 @@ jobs:
56
# Install the CUDA 12.8 toolkit (nvcc + cudart headers) into the build
57
# container once. Script lives in .github/ (scripts/ is .gitignore'd).
58
CIBW_BEFORE_ALL_LINUX: bash {project}/.github/install_cuda_el8.sh 12-8
59
+ # NB: quote the arch list — cibuildwheel parses CIBW_ENVIRONMENT with
60
+ # bashlex, so unquoted ';' would be read as a command separator and
61
+ # error out ("Malformed environment") before any build starts.
62
CIBW_ENVIRONMENT_LINUX: >
63
SWEEP_BUILD_CUDA=1
- TORCH_CUDA_ARCH_LIST=7.0;7.5;8.0;8.6;8.9;9.0
64
+ TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6;8.9;9.0"
65
CUDA_HOME=/usr/local/cuda
66
PATH=/usr/local/cuda/bin:$PATH
67
# Build against the exact torch users resolve at install time.
0 commit comments