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 ceade38 commit a569f6bCopy full SHA for a569f6b
cmake/Cuda.cmake
@@ -113,8 +113,8 @@ function(caffe_select_nvcc_arch_flags out_variable)
113
set(__nvcc_flags "")
114
set(__nvcc_archs_readable "")
115
116
- string(COMPARE GREATER_EQUAL "${CUDA_VERSION}" "9.1" iscudanewerthan91)
117
- if(iscudanewerthan91)
+ string(COMPARE GREATER_EQUAL "${CUDA_VERSION}" "9.0" iscudanewerthan90)
+ if(iscudanewerthan90)
118
string(REPLACE "21(20)" "" __cuda_arch_bin "${__cuda_arch_bin}")
119
string(REPLACE "20" "" __cuda_arch_bin "${__cuda_arch_bin}")
120
endif()
0 commit comments