文章目录
前言
看网上的博客,都说编译tf_ops,tensorflow_gpu的版本在1.13以下会比较容易,但我就是要用1.15,就是开心,就是玩!
1.14以上不能编译的原因
1.14版本无法解决其实是 因为编译的时候链接文件为libtensorflow_framework.so, 而tensorflow1.14及之后的版本并不存在这个文件,而是libtensorflow_framework.so.1(tensorflow2.0我没有安装过,但是听说也没有那个文件)。这个问题其实很好解决,只要生成一个libtensorflow_framework.so.1的软链接就可以。【我没有用到软连接,可能是tensorflow官方修改了?(2021.9.8)】
创建虚拟环境
conda create -n pointnet2 python=3.6
conda activate pointnet2
在conda虚拟环境中安装cuda、cudnn、tensorflow_gpu-1.15
版本对应可参考这里
conda install cudatoolkit=9.2.0 cudnn=7.2.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
pip install tensorflow-gpu==1.5.0
查看conda环境下的cuda和cundnn版本
conda list cuda
conda list cudnn
安装一些包
pip install scipy
pip install h5py
pip install open3d-python
出错
AttributeError: module 'tensorflow.python.platform.sysconfig' has no attribute 'get_compile_flags'
解决
编译tf_ops
编译之前
前面说了,如果安装tf1.13以上的话,需要增加一点操作.