Skip to content

Commit 7d5e98f

Browse files
authored
Add CUDA 11.6 version to manywheel script
Make sure TORCH_CUDA_ARCH_LIST and EXTRA_CAFFE2_CMAKE_FLAGS are set for CUDA 11.6
1 parent 96a30f6 commit 7d5e98f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: manywheel/build_cuda.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ cuda_version_nodot=$(echo $CUDA_VERSION | tr -d '.')
5858

5959
TORCH_CUDA_ARCH_LIST="3.7;5.0;6.0;7.0"
6060
case ${CUDA_VERSION} in
61-
11.[35])
61+
11.[356])
6262
TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST};7.5;8.0;8.6"
6363
EXTRA_CAFFE2_CMAKE_FLAGS+=("-DATEN_NO_TEST=ON")
6464
;;

0 commit comments

Comments
 (0)