tf编译pointnet2-master中sampling层

本文主要是介绍tf编译pointnet2-master中sampling层，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

1 .\tf_ops\sampling\文件夹下

打开tf_sampling_compile.sh

修改几个参数

#/bin/bash
/usr/local/cuda/bin/nvcc tf_sampling_g.cu -o tf_sampling_g.cu.o -c -O2 -DGOOGLE_CUDA=1 -x cu -Xcompiler -fPICTF_ROOT=./venv/lib/python3.6/site-packages/tensorflow
# TF1.2
#g++ -std=c++11 tf_sampling.cpp tf_sampling_g.cu.o -o tf_sampling_so.so -shared -fPIC -I ${TF_ROOT}/include -I /usr/local/cuda/include -lcudart -L /usr/local/cuda/lib64/ -O2 -D_GLIBCXX_USE_CXX11_ABI=0# TF1.4
g++ -std=c++11 tf_sampling.cpp tf_sampling_g.cu.o -o tf_sampling_so.so -shared -fPIC -I ${TF_ROOT}/include -I /usr/local/cuda/include -I ${TF_ROOT}/include/external/nsync/public -lcudart -L /usr/local/cuda/lib64/ -L ${TF_ROOT}  -ltensorflow_framework -O2 -D_GLIBCXX_USE_CXX11_ABI=0

添加一个

这篇关于tf编译pointnet2-master中sampling层的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---