@@ -13,6 +13,8 @@ if [ ! "$DEEPDETECT_CUDA_ARCH" ]; then
13
13
for card in 30 35 50 52 61 62 70 72; do
14
14
DEEPDETECT_CUDA_ARCH=" $DEEPDETECT_CUDA_ARCH -gencode arch=compute_${card} ,code=sm_${card} "
15
15
done
16
+ # trim spaces
17
+ DEEPDETECT_CUDA_ARCH=" $( echo ${DEEPDETECT_CUDA_ARCH} | xargs) "
16
18
fi
17
19
18
20
@@ -144,22 +146,22 @@ gpu_build() {
144
146
case ${DEEPDETECT_BUILD} in
145
147
146
148
" tf" )
147
- cmake .. -DUSE_TF=ON -DUSE_CUDNN=ON -DUSE_XGBOOST=ON -DUSE_SIMSEARCH=ON -DUSE_TSNE=ON -DCUDA_ARCH=${DEEPDETECT_CUDA_ARCH}
149
+ cmake .. -DUSE_TF=ON -DUSE_CUDNN=ON -DUSE_XGBOOST=ON -DUSE_SIMSEARCH=ON -DUSE_TSNE=ON -DCUDA_ARCH=" ${DEEPDETECT_CUDA_ARCH} "
148
150
make
149
151
;;
150
152
151
153
" caffe-tf-cpu" )
152
- cmake .. -DUSE_TF=ON -DUSE_TF_CPU_ONLY=ON -DUSE_CUDNN=ON -DUSE_XGBOOST=ON -DUSE_SIMSEARCH=ON -DUSE_TSNE=ON -DCUDA_ARCH=${DEEPDETECT_CUDA_ARCH}
154
+ cmake .. -DUSE_TF=ON -DUSE_TF_CPU_ONLY=ON -DUSE_CUDNN=ON -DUSE_XGBOOST=ON -DUSE_SIMSEARCH=ON -DUSE_TSNE=ON -DCUDA_ARCH=" ${DEEPDETECT_CUDA_ARCH} "
153
155
make
154
156
;;
155
157
156
158
" caffe-tf" )
157
- cmake .. -DUSE_TF=ON -DUSE_CUDNN=ON -DUSE_SIMSEARCH=ON -DUSE_TSNE=ON -DCUDA_ARCH=${DEEPDETECT_CUDA_ARCH}
159
+ cmake .. -DUSE_TF=ON -DUSE_CUDNN=ON -DUSE_SIMSEARCH=ON -DUSE_TSNE=ON -DCUDA_ARCH=" ${DEEPDETECT_CUDA_ARCH} "
158
160
make
159
161
;;
160
162
161
163
" caffe2" )
162
- cmake .. -DUSE_CUDNN=ON -DUSE_XGBOOST=ON -DUSE_SIMSEARCH=ON -DUSE_TSNE=ON -DUSE_CAFFE2=ON -DCUDA_ARCH=${DEEPDETECT_CUDA_ARCH}
164
+ cmake .. -DUSE_CUDNN=ON -DUSE_XGBOOST=ON -DUSE_SIMSEARCH=ON -DUSE_TSNE=ON -DUSE_CAFFE2=ON -DCUDA_ARCH=" ${DEEPDETECT_CUDA_ARCH} "
163
165
make
164
166
;;
165
167
@@ -179,12 +181,12 @@ gpu_build() {
179
181
;;
180
182
181
183
" faiss" )
182
- cmake .. -DUSE_CUDNN=ON -DUSE_FAISS=ON -DUSE_XGBOOST=ON -DUSE_SIMSEARCH=ON -DUSE_TSNE=ON -DCUDA_ARCH=${DEEPDETECT_CUDA_ARCH}
184
+ cmake .. -DUSE_CUDNN=ON -DUSE_FAISS=ON -DUSE_XGBOOST=ON -DUSE_SIMSEARCH=ON -DUSE_TSNE=ON -DCUDA_ARCH=" ${DEEPDETECT_CUDA_ARCH} "
183
185
make
184
186
;;
185
187
186
188
* )
187
- cmake .. -DUSE_CUDNN=ON -DUSE_XGBOOST=ON -DUSE_SIMSEARCH=ON -DUSE_TSNE=ON -DCUDA_ARCH=${DEEPDETECT_CUDA_ARCH}
189
+ cmake .. -DUSE_CUDNN=ON -DUSE_XGBOOST=ON -DUSE_SIMSEARCH=ON -DUSE_TSNE=ON -DCUDA_ARCH=" ${DEEPDETECT_CUDA_ARCH} "
188
190
make
189
191
;;
190
192
esac
0 commit comments