mirror of
https://github.com/wassname/pyrobolearn.git
synced 2026-09-09 11:31:38 +08:00
add license, requirements, setup and install scripts
This commit is contained in:
Executable
+94
@@ -0,0 +1,94 @@
|
||||
#!/bin/sh
|
||||
# This bash script install the openpose and the official python wrapper
|
||||
#
|
||||
# Github repo: https://github.com/CMU-Perceptual-Computing-Lab/openpose
|
||||
# Reference: https://github.com/CMU-Perceptual-Computing-Lab/openpose/blob/master/doc/installation.md
|
||||
# FAQ (troubleshooting): https://github.com/CMU-Perceptual-Computing-Lab/openpose/blob/master/doc/faq.md
|
||||
|
||||
|
||||
# Define few variables
|
||||
currentdir=$PWD
|
||||
|
||||
# Check if user is root/running with sudo
|
||||
if [ `whoami` != root ]; then
|
||||
echo "Please run this script with sudo"
|
||||
exit
|
||||
fi
|
||||
|
||||
# check the first argument which must be the source directory
|
||||
if [ -z $1 ]
|
||||
then
|
||||
echo "Specify the absolute location (or relative to home) where to install the Openpose package."
|
||||
exit
|
||||
fi
|
||||
|
||||
# move to that directory
|
||||
cd; cd $1
|
||||
|
||||
# clone repository
|
||||
git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose
|
||||
cd openpose
|
||||
git pull origin master
|
||||
|
||||
# Prerequisites
|
||||
sudo apt-get install cmake-qt-gui
|
||||
sudo apt install cmake-curses-gui
|
||||
read -p 'Do you wish to install the CUDA8 on this computer [yes/no]?: ' reply
|
||||
if [ $reply = 'yes' ]; then
|
||||
echo "Installing CUDA8..."
|
||||
sudo ./scripts/ubuntu/install_cuda.sh
|
||||
else
|
||||
echo "Not installing CUDA8"
|
||||
fi
|
||||
read -p 'Do you wish to install the cuDNN5.1 on this computer [yes/no]?: ' reply
|
||||
if [ $reply = 'yes' ]; then
|
||||
echo "Installing cuDNN5.1..."
|
||||
sudo ./scripts/ubuntu/install_cudnn.sh
|
||||
else
|
||||
echo "Not installing cuDNN5.1"
|
||||
fi
|
||||
sudo ./scripts/ubuntu/install_deps.sh
|
||||
sudo apt-get install libopencv-dev
|
||||
sudo apt-get install nvidia-modprobe
|
||||
|
||||
mkdir build && cd build
|
||||
echo "Check https://github.com/CMU-Perceptual-Computing-Lab/openpose/blob/master/doc/installation.md#openpose-configuration for more information on how to configure the openpose package"
|
||||
#cmake-gui
|
||||
# to build with python 2.7
|
||||
#export PYTHON_EXECUTABLE=/usr/bin/python2.7
|
||||
#export PYTHON_LIBRARY=/usr/lib/x86_64-linux-gnu/libpython2.7.so
|
||||
cmake -DBUILD_PYTHON=ON -DPYBIND11_PYTHON_VERSION=2.7 ..
|
||||
make -j`nproc`
|
||||
sudo make install
|
||||
sudo cp python/openpose/pyopenpose.so /usr/local/lib/python2.7/dist-packages/
|
||||
cd ..
|
||||
OPENPOSE_PATH_DIR=$PWD
|
||||
|
||||
# go to home directory and export environment variables in the bashrc
|
||||
cd
|
||||
echo "\n\n# Export OPENPOSE environment variables" >> ".bashrc"
|
||||
echo "export OPENPOSE_PATH=$OPENPOSE_PATH_DIR" >> ".bashrc"
|
||||
|
||||
# Test installation (check github.com/CMU-Perceptual-Computing-Lab/openpose/blob/master/doc/quick_start.md)
|
||||
#cd OPENPOSE_PATH_DIR
|
||||
#./build/examples/openpose/openpose.bin
|
||||
# Python examples are in github.com/CMU-Perceptual-Computing-Lab/openpose/tree/master/examples/tutorial_api_python
|
||||
|
||||
# return to original directory
|
||||
cd currentdir
|
||||
|
||||
|
||||
# Troubleshootings
|
||||
# - If you obtain the following error "nvcc fatal : Unsupported gpu architecture 'compute_<xx>'", just modify the
|
||||
# cmake/Cuda.cmake file, from the line "set(Caffe_known_gpu_archs "...")", remove the architectures that your
|
||||
# computer does not possess. You can also specify the architecture in the cmake line with the option "CUDA_ARCH".
|
||||
# - If you obtain an error with the python version, you have to specify the version you want to use to pybind11, by
|
||||
# setting the option "-DPYBIND11_PYTHON_VERSION=2.7"
|
||||
# - If you have the following error "Cuda check failed (30 vs. 0): unknown error" when launching a basic example
|
||||
# in openpose as given in "https://github.com/CMU-Perceptual-Computing-Lab/openpose/blob/master/doc/quick_start.md",
|
||||
# then the error comes from your Nvidia GPU driver or your CUDA toolkit. Check the solutions given in the following
|
||||
# link: https://github.com/NVIDIA/DIGITS/issues/1663
|
||||
# Before tempting any solutions, try to reboot your computer (this solution worked for me). Otherwise, this will
|
||||
# probably require you to reinstall the Nvidia drivers / CUDA toolkit.
|
||||
# - If you have an error with Qt such as: qt.qpa.plugin: Could not find the Qt platform plugin "xcb" in "", reinstall
|
||||
# the package, or install qtcreator.
|
||||
Reference in New Issue
Block a user