8000 [BE] cleanup install_cuda.sh script (#1942) · pytorch/builder@e7b7315 · GitHub
[go: up one dir, main page]

Skip to content

Commit e7b7315

Browse files
authored
[BE] cleanup install_cuda.sh script (#1942)
1 parent 1512b5f commit e7b7315

File tree

2 files changed

+0
-253
lines changed

2 files changed

+0
-253
lines changed

common/install_cuda.sh

Lines changed: 0 additions & 239 deletions
This file was deleted.

manywheel/build_cuda.sh

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -85,20 +85,6 @@ case ${CUDA_VERSION} in
8585
;;
8686
esac
8787

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-
10288
export TORCH_CUDA_ARCH_LIST=${TORCH_CUDA_ARCH_LIST}
10389
echo "${TORCH_CUDA_ARCH_LIST}"
10490

0 commit comments

Comments
 (0)
0