3记录一下openpose tf版本的安装全流程,踩了所有的坑,希望你可以一次性安装成功。
环境:Ubuntu16.04+Anaconda3
code:https://github.com/ildoonet/tf-pose-estimation
paper:https://arxiv.org/pdf/1611.08050.pdf
1)创建虚拟环境+安装cuda+cudnn+tensorflow-gpu:
首先创建一个python3.6的虚拟环境,设置conda国内镜像源,然后使用conda来安装tensorflow-gpu,conda安装的好处会自动装上一系列依赖,包括cuda和cudnn(这个前提是安装好了对应显卡的驱动,如何安装驱动有很多教程就不赘述。比起在官网自己下载设置cuda和cudnn,使用conda安装,非常方便。) 会自动安装好tensorflow-gpu 1.4.1所对应的cuda和cudnn版本。
conda create -n openposetf python=3.6
source activate openposetf
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/