We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1512b5f commit fe908e1Copy full SHA for fe908e1
2 files changed
common/install_cuda.sh
manywheel/build_cuda.sh
@@ -85,20 +85,6 @@ case ${CUDA_VERSION} in
85
;;
86
esac
87
88
-if [[ -n "$OVERRIDE_TORCH_CUDA_ARCH_LIST" ]]; then
89
- TORCH_CUDA_ARCH_LIST="$OVERRIDE_TORCH_CUDA_ARCH_LIST"
90
-
91
- # Prune CUDA again with new arch list. Unfortunately, we need to re-install CUDA to prune it again
92
- override_gencode=""
93
- for arch in ${TORCH_CUDA_ARCH_LIST//;/ } ; do
94
- arch_code=$(echo "$arch" | tr -d .)
95
- override_gencode="${override_gencode}-gencode arch=compute_$arch_code,code=sm_$arch_code "
96
- done
97
98
- export OVERRIDE_GENCODE=$override_gencode
99
- bash "$(dirname "$SCRIPTPATH")"/common/install_cuda.sh "${CUDA_VERSION}"
100
-fi
101
102
export TORCH_CUDA_ARCH_LIST=${TORCH_CUDA_ARCH_LIST}
103
echo "${TORCH_CUDA_ARCH_LIST}"
104
0 commit comments