Frustum PoitNets的正确编译生成tf_op的.so文件

基于Colab环境下的Frustum PoitNets论文代码复现,遇到的tf_op生成问题,其实就是这三个文件的生成:

3d_interpolation.so

grouping.so

sampling.so

论文作者代码中的文件适用于当时开发时的环境,但不一定适用于所有复现论文的人。因此在colab平台,试出了这么一个环境能够正确生成.so文件。

  • Ubuntu 18.04.5 LTS
  • tensorflow-gpu==1.13.1
  • numpy==1.16.0 (this might be the key)
  • gcc (Ubuntu 5.5.0-12ubuntu1) 5.5.0
  • g++ (Ubuntu 7.5.0-3ubuntu1~18.04) 7.5.0
  • CUDA 10.0
  • And "-D_GLIBCXX_USE_CXX11_ABI=0" in the "tf_xxxx_complie.sh" should be deleted

在colab可以通过运行以下代码:

!python3.7 -m pip uninstall tensorflow
!python3.7 -m pip install tensorflow-gpu==1.13.1
!python3.7 -m pip install scipy
!python3.7 -m pip install numpy==1.16.0
!python3.7 -m pip install mayavi

!gcc --version
!g++ --version
!whereis gcc-5
!sudo apt-get install gcc-5
!--install    
!sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 100
!sudo apt autoremove
!sudo apt-get update gcc-5
!sudo apt-get install g++-5
!gcc --version
!g++ --version
!whereis gcc-5


import os
os.chdir("/content/drive/MyDrive/frustum/frustum_floders/codes/frustum-pointnets-master/models/tf_ops/3d_interpolation")
!sh tf_interpolate_compile.sh
os.chdir("/content/drive/MyDrive/frustum/frustum_floders/codes/frustum-pointnets-master/models/tf_ops/grouping")
!sh tf_grouping_compile.sh
os.chdir("/content/drive/MyDrive/frustum/frustum_floders/codes/frustum-pointnets-master/models/tf_ops/sampling")
!sh tf_sampling_compile.sh

文件夹的位置需要根据云端硬盘的储存方式更改一下

你可能感兴趣的:(深度学习,人工智能,python)