File tree Expand file tree Collapse file tree 2 files changed +0
-253
lines changed Expand file tree Collapse file tree 2 files changed +0
-253
lines changed Load Diff This file was deleted.
Original file line number Diff line number Diff line change @@ -85,20 +85,6 @@ case ${CUDA_VERSION} in
85
85
;;
86
86
esac
87
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
88
export TORCH_CUDA_ARCH_LIST=${TORCH_CUDA_ARCH_LIST}
103
89
echo " ${TORCH_CUDA_ARCH_LIST} "
104
90
You can’t perform that action at this time.
0 commit comments