ubuntu1604_python3.5_cuda10_cudnn7.5安装pytorch、mxnet、openpose配置文档
- 系统概述
- ubuntu1604
- cuda10
- cudnn7.5
- 安装pytorch-gpu
- 下载pytorch-gpu版本,下载地址:https://pytorch.org/
- 选择“Stable1.2->Linux->pip->python3.5->cuda10”。如下图所示:
说明:下载前先确认自己的系统、python版本、cuda版本,必须一一匹配。否则安装会出现各种问题。
我下载的版本是:torch-1.2.0-cp35-cp35m-manylinux1_x86_64.whl
- 在终端输入pip3 install torch torchvision 进行安装。
说明:安装过程如果有报错,可以自行百度,百度上可以搜出解决方法。这里不再一一举例了。
pytorch安装成功的标志如下图所示:
import torch无报错,说明安装成功。
- 安装mxnet-gpu
- 在终端输入:pip3 install mxnet-cu100
说明:如果安装的cuda8、或者cuda9,需要下载与cuda版本对应的mxnet
2.cuda10下载的版本为:
mxnet_cu100-1.5.0-py2.py3-none-manylinux1_x86_64.whl
- mxnet安装成功的标志如下图所示:
import mxnet无报错,说明安装成功。
- 安装过程中的报错,解决情况说明:
- import mxnet报错:ImportError: libgfortran.so.3: cannot open shared object file: No such file or directory
解决方法:sudo apt-get install libgfortran3
- 安装openpose
- 下载openpose
- 在终端输入: git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose.git
- 安装Cmake Gui: sudo apt-get install cmake-gui
- 在终端输入
- cd openpose
- cd models
- ./getModels.sh (下载模型时间会比较长,请耐心等待……)
模型下载完成后,在openpose目录创建build文件夹,打开终端输入
- sudo mkdir build
- 启动cmake-gui进行编译,在终端输入:
- cmake-gui
- 弹出界面如下:
填写说明:
- where is the source code:/home/yd/openpose
- where to build the binaries :/home/yd/openpose/build
- Caffe_INCLUDE_DIRS:/home/yd/openpose/3rdparty/caffe/include
- Caffe_LIBS: /home/yd/openpose/3rdparty/caffe/build/lib/libcaffe.so
以上路径请根据自己电脑的实际情况进行填写。
- 编译openpose
- cd build/
- sudo cmake ..
- sudo make -j8
- 编译过程中,如果100%没有报错的话,基本就编译成功了,在openpose目录下,输入命令:
./build/examples/openpose/openpose.bin --video examples/media/video.avi
显示下图,即表示安装成功。