We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8ffd0f9 commit 52c2f25Copy full SHA for 52c2f25
manywheel/build.sh
@@ -83,7 +83,7 @@ if [[ -n "$OVERRIDE_TORCH_CUDA_ARCH_LIST" ]]; then
83
84
# Prune CUDA again with new arch list. Unfortunately, we need to re-install CUDA to prune it again
85
override_gencode=""
86
- for arch in ${torch_cuda_arch_list//;/ } ; do
+ for arch in ${TORCH_CUDA_ARCH_LIST//;/ } ; do
87
arch_code=$(echo "$arch" | tr -d .)
88
override_gencode="${override_gencode}-gencode arch=compute_$arch_code,code=sm_$arch_code "
89
done
0 commit comments