forked from dvlab-research/3DSSD
-
Notifications
You must be signed in to change notification settings - Fork 0
/
compile_all.sh
executable file
·46 lines (38 loc) · 2.35 KB
/
compile_all.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
# evaluation
TFPATH=$1 # e.g: /usr/local/lib/python3.5/dist-packages/tensorflow
CUDAPATH=$2 # e.g: /usr/local/cuda-9.0
OPSPATH="lib/utils/tf_ops"
# voxel operation
cd lib/builder/voxel_generator
./build.sh
cd dist
pip install points2voxel-0.0.1-cp36-cp36m-linux_x86_64.whl
cd ../../../..
# evaluation
cd ${OPSPATH}/evaluation
/usr/bin/gcc-5 -std=c++11 tf_evaluate.cpp -o tf_evaluate_so.so -shared -fPIC -I ${TFPATH}/include -I ${CUDAPATH}/include -I ${TFPATH}/include/external/nsync/public -lcudart -L ${CUDAPATH}/lib64/ -L ${TFPATH} -ltensorflow_framework -O2 -D_GLIBCXX_USE_CXX11_ABI=0
cd ..
# grouping
cd grouping
$CUDAPATH/bin/nvcc -ccbin /usr/bin/gcc-5 tf_grouping_g.cu -o tf_grouping_g.cu.o -c -O2 -DGOOGLE_CUDA=1 -x cu -Xcompiler -fPIC
/usr/bin/gcc-5 -std=c++11 tf_grouping.cpp tf_grouping_g.cu.o -o tf_grouping_so.so -shared -fPIC -I $TFPATH/include -I $CUDAPATH/include -I $TFPATH/include/external/nsync/public -lcudart -L $CUDAPATH/lib64/ -L $TFPATH -ltensorflow_framework -O2 -D_GLIBCXX_USE_CXX11_ABI=0
cd ..
# interpolation
cd interpolation
$CUDAPATH/bin/nvcc -ccbin /usr/bin/gcc-5 tf_interpolate_g.cu -o tf_interpolate_g.cu.o -c -O2 -DGOOGLE_CUDA=1 -x cu -Xcompiler -fPIC
/usr/bin/gcc-5 -std=c++11 tf_interpolate.cpp tf_interpolate_g.cu.o -o tf_interpolate_so.so -shared -fPIC -I $TFPATH/include -I $CUDAPATH/include -I $TFPATH/include/external/nsync/public -lcudart -L $CUDAPATH/lib64/ -L $TFPATH -ltensorflow_framework -O2 -D_GLIBCXX_USE_CXX11_ABI=0
cd ..
# points pooling
cd points_pooling
$CUDAPATH/bin/nvcc -ccbin /usr/bin/gcc-5 tf_points_pooling_g.cu -o tf_points_pooling_g.cu.o -c -O2 -DGOOGLE_CUDA=1 -x cu -Xcompiler -fPIC
/usr/bin/gcc-5 -std=c++11 tf_points_pooling.cpp tf_points_pooling_g.cu.o -o tf_points_pooling_so.so -shared -fPIC -I $TFPATH/include -I $CUDAPATH/include -I $TFPATH/include/external/nsync/public -lcudart -L $CUDAPATH/lib64/ -L $TFPATH -ltensorflow_framework -O2 -D_GLIBCXX_USE_CXX11_ABI=0
cd ..
# sampling
cd sampling
$CUDAPATH/bin/nvcc -ccbin /usr/bin/gcc-5 tf_sampling_g.cu -o tf_sampling_g.cu.o -c -O2 -DGOOGLE_CUDA=1 -x cu -Xcompiler -fPIC
/usr/bin/gcc-5 -std=c++11 tf_sampling.cpp tf_sampling_g.cu.o -o tf_sampling_so.so -shared -fPIC -I $TFPATH/include -I $CUDAPATH/include -I $TFPATH/include/external/nsync/public -lcudart -L $CUDAPATH/lib64/ -L $TFPATH -ltensorflow_framework -O2 -D_GLIBCXX_USE_CXX11_ABI=0
cd ..
# nms
cd nms
./build.sh
cd ..