Skip to content

Commit 696087f

Browse files
committed
fix(build): CUDA_ARCH not escaped correctly
1 parent 67ad43e commit 696087f

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

build.sh

+8-6
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,8 @@ if [ ! "$DEEPDETECT_CUDA_ARCH" ]; then
1313
for card in 30 35 50 52 61 62 70 72; do
1414
DEEPDETECT_CUDA_ARCH="$DEEPDETECT_CUDA_ARCH -gencode arch=compute_${card},code=sm_${card}"
1515
done
16+
# trim spaces
17+
DEEPDETECT_CUDA_ARCH="$(echo ${DEEPDETECT_CUDA_ARCH} | xargs)"
1618
fi
1719

1820

@@ -144,22 +146,22 @@ gpu_build() {
144146
case ${DEEPDETECT_BUILD} in
145147

146148
"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}"
148150
make
149151
;;
150152

151153
"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}"
153155
make
154156
;;
155157

156158
"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}"
158160
make
159161
;;
160162

161163
"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}"
163165
make
164166
;;
165167

@@ -179,12 +181,12 @@ gpu_build() {
179181
;;
180182

181183
"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}"
183185
make
184186
;;
185187

186188
*)
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}"
188190
make
189191
;;
190192
esac

0 commit comments

Comments
 (0)