|
1 | 1 | # Python Training Library Installation |
2 | 2 |
|
3 | | -## Prerequisites |
4 | | -* [Anaconda3](https://www.anaconda.com/products/individual):<br> |
5 | | - Anaconda is used to create virtual environment that facilitates building the running environment and ease the complexity of library depedencies. Here we mainly use it to create virtual python environment and install cuda run-time libraries. |
6 | | -* [CUDA](https://developer.nvidia.com/cuda-downloads):<br> |
7 | | - CUDA enviroment is essential to run deep learning neural networks on GPUs. The CUDA installation packages to download should match your system and your NVIDIA Driver version. |
| 3 | +## Configure CUDA environment |
8 | 4 |
|
9 | | -## Configure environment |
10 | | - There are two ways to install hyperpose python training library. |
| 5 | +You can configure your CUDA either by Anaconda or your system setting. |
11 | 6 |
|
12 | | - All the following instructions have been tested on the environments below:<br> |
13 | | -| OS | NVIDIA Driver | CUDA Toolkit | GPU | |
14 | | -| ------------ | ------------- | ------------ | -------------- | |
15 | | -| Ubuntu 18.04 | 410.79 | 10.0 | Tesla V100-DGX | |
16 | | -| Ubuntu 18.04 | 440.33.01 | 10.2 | Tesla V100-DGX | |
17 | | -| Ubuntu 18.04 | 430.64 | 10.1 | TITAN RTX | |
18 | | -| Ubuntu 18.04 | 430.26 | 10.2 | TITAN XP | |
19 | | -| Ubuntu 16.04 | 430.50 | 10.1 | RTX 2080Ti | |
| 7 | +### Using CUDA toolkits from Anaconda (RECOMMENDED) |
| 8 | + |
| 9 | +:::{admonition} Prerequisites |
| 10 | +- [Anaconda3](https://www.anaconda.com/products/individual) |
| 11 | +- [NVidia Driver >= 410.79 (required by CUDA 10)](https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#driver-installation) |
| 12 | +::: |
20 | 13 |
|
| 14 | +It is suggested to create new conda environment regarding the CUDA requirements. |
21 | 15 |
|
22 | | - Before all, we recommend you to create anaconda virtual environment first, which could handle the possible conflicts between the libraries you already have in your computers and the libraries hyperpose need to install, and also handle the dependencies of the cudatoolkit and cudnn library in a very simple way.<br> |
23 | | - To create the virtual environment, run the following command in bash: |
24 | 16 | ```bash |
25 | | -# >>> create virtual environment (choose yes) |
26 | | -conda create -n hyperpose python=3.7 |
| 17 | +# >>> create virtual environment |
| 18 | +conda create -n hyperpose python=3.7 -y |
27 | 19 | # >>> activate the virtual environment, start installation |
28 | 20 | conda activate hyperpose |
29 | 21 | # >>> install cudatoolkit and cudnn library using conda |
30 | 22 | conda install cudatoolkit=10.0.130 |
31 | 23 | conda install cudnn=7.6.0 |
32 | 24 | ``` |
| 25 | + |
| 26 | +::::{warning} |
| 27 | +It is also possible to install CUDA dependencies without creating a new environment. |
| 28 | +But it might introduce environment conflicts. |
| 29 | + |
| 30 | +:::{code-block} bash |
| 31 | +conda install cudatoolkit=10.0.130 |
| 32 | +conda install cudnn=7.6.0 |
| 33 | +::: |
| 34 | +:::: |
| 35 | + |
| 36 | + |
| 37 | +### Using system-wide CUDA toolkits |
| 38 | + |
| 39 | +Users may also directly depend on the system-wide CUDA and CuDNN libraries. |
| 40 | + |
| 41 | +HyperPose have been tested on the environments below: |
| 42 | + |
| 43 | +| OS | NVIDIA Driver | CUDA Toolkit | GPU | |
| 44 | +| ------------ | ------------- | ------------ | -------------- | |
| 45 | +| Ubuntu 18.04 | 410.79 | 10.0 | Tesla V100-DGX | |
| 46 | +| Ubuntu 18.04 | 440.33.01 | 10.2 | Tesla V100-DGX | |
| 47 | +| Ubuntu 18.04 | 430.64 | 10.1 | TITAN RTX | |
| 48 | +| Ubuntu 18.04 | 430.26 | 10.2 | TITAN XP | |
| 49 | +| Ubuntu 16.04 | 430.50 | 10.1 | RTX 2080Ti | |
| 50 | + |
| 51 | +::::{admonition} Check CUDA/CuDNN versions |
33 | 52 |
|
34 | | - After configuring and activating conda enviroment, we can then begin to install the hyperpose.<br> |
| 53 | +To test CUDA version, run `nvcc --version`: the highlight line in the output indicates that you have CUDA 11.2 installed. |
| 54 | +:::{code-block} bash |
| 55 | +:emphasize-lines: 5 |
| 56 | +nvcc --version |
| 57 | +# ========== Valid output looks like ========== |
| 58 | +# nvcc: NVIDIA (R) Cuda compiler driver |
| 59 | +# Copyright (c) 2005-2020 NVIDIA Corporation |
| 60 | +# Built on Mon_Nov_30_19:08:53_PST_2020 |
| 61 | +# Cuda compilation tools, release 11.2, V11.2.67 |
| 62 | +# Build cuda_11.2.r11.2/compiler.29373293_0 |
| 63 | +::: |
35 | 64 |
|
36 | | -### (I)The first method to install is to put hyperpose python module in the working directory.(recommand)<br> |
37 | | - After git-cloning the source [repository](https://github.com/tensorlayer/hyperpose.git), you can directly import hyperpose python library under the root directory of the cloned repository.<br> |
| 65 | +To check your system-wide CuDNN version **on Linux**: the output (in the comment) shows that we have CuDNN 8.0.5. |
| 66 | +:::{code-block} bash |
| 67 | +ls /usr/local/cuda/lib64 | grep libcudnn.so |
| 68 | +# === Valid output looks like === |
| 69 | +# libcudnn.so |
| 70 | +# libcudnn.so.8 |
| 71 | +# libcudnn.so.8.0.5 |
| 72 | +::: |
| 73 | +:::: |
38 | 74 |
|
39 | | - To make importion available, you should install the prerequist dependencies as followed:<br> |
40 | | - you can either install according to the requirements.txt in the [repository](https://github.com/tensorlayer/hyperpose.git) |
| 75 | +## Install HyperPose Python training library |
| 76 | + |
| 77 | +### Install with `pip` |
| 78 | + |
| 79 | +To install a stable library from [Python Package Index](https://github.com/tensorlayer/hyperpose): |
41 | 80 |
|
42 | 81 | ```bash |
43 | | -# install according to the requirements.txt |
44 | | -pip install -r requirements.txt |
| 82 | +pip install -U hyperpose |
45 | 83 | ``` |
46 | 84 |
|
47 | | - or install libraries one by one |
| 85 | +Or you can install a specific release of hyperpose from GitHub, for example: |
48 | 86 |
|
49 | 87 | ```bash |
50 | | -# >>> install tensorflow of version 2.3.1 |
51 | | -pip install tensorflow-gpu==2.3.1 |
52 | | -# >>> install tensorlayer of version 2.2.3 |
53 | | -pip install tensorlayer==2.2.3 |
54 | | -# >>> install other requirements (numpy<=17.0.0 because it has conflicts with pycocotools) |
55 | | -pip install opencv-python |
56 | | -pip install numpy==1.16.4 |
57 | | -pip install pycocotools |
58 | | -pip install matplotlib |
| 88 | +export HYPERPOSE_VERSION="2.2.0-alpha" |
| 89 | +pip install -U https://github.com/tensorlayer/hyperpose/archive/${HYPERPOSE_VERSION}.zip |
59 | 90 | ``` |
60 | 91 |
|
61 | | - This method of installation use the latest source code and thus is less likely to meet compatibility problems.<br><br> |
| 92 | +More GitHub releases and its version can be found [here](https://github.com/tensorlayer/hyperpose/releases). |
62 | 93 |
|
63 | | -### (II)The second method to install is to use pypi repositories.<br> |
64 | | - We have already upload hyperpose python library to pypi website so you can install it using pip, which gives you the last stable version. |
| 94 | +### Local installation |
| 95 | + |
| 96 | +You can also install HyperPose by installing the raw GitHub repository, this is usually for developers. |
65 | 97 |
|
66 | 98 | ```bash |
67 | | - pip install hyperpose |
| 99 | +# Install the source codes from GitHub |
| 100 | +git clone https://github.com/tensorlayer/hyperpose.git |
| 101 | +pip install -U -r hyperpose/requirements.txt |
| 102 | + |
| 103 | +# Add `hyperpose/hyperpose` to `PYTHONPATH` to help python find it. |
| 104 | +export HYPERPOSE_PYTHON_HOME=$(pwd)/hyperpose |
| 105 | +export PYTHONPATH=$HYPERPOSE_PYTHON_HOME/python:${PYTHONPATH} |
68 | 106 | ``` |
69 | 107 |
|
70 | | - This will download and install all dependencies automatically. |
| 108 | +## Check the installation |
| 109 | + |
| 110 | +Let's check whether HyperPose is installed by running following commands: |
71 | 111 |
|
72 | | - Now after installing dependent libraries and hyperpose itself, let's check whether the installation successes. |
73 | | -run following command in bash: |
74 | 112 | ```bash |
75 | | -# >>> now the configuration is done, check whether the GPU is avaliable. |
76 | | -python |
77 | | ->>> import tensorflow as tf |
78 | | ->>> import tensorlayer as tl |
79 | | ->>> tf.test.is_gpu_available() |
80 | | -# >>> if the output is True, congratulation! you can import and run hyperpose now |
81 | | ->>> from hyperpose import Config,Model,Dataset |
| 113 | +python -c ' |
| 114 | +import tensorflow as tf # Test TensorFlow installation |
| 115 | +import tensorlayer as tl # Test TensorLayer installation |
| 116 | +assert tf.test.is_gpu_available() # Test GPU availability |
| 117 | +import hyperpose # Test HyperPose import |
| 118 | +' |
82 | 119 | ``` |
83 | 120 |
|
84 | | -## Extra configuration for exporting model |
85 | | - The hypeprose python training library handles the whole pipelines for developing the pose estimation system, including training, evaluating and testing. Its goal is to produce a **.npz** file that contains the well-trained model weights. |
| 121 | +## Optional Setup |
| 122 | + |
| 123 | +### Extra configurations for exporting models |
86 | 124 |
|
87 | | - For the training platform, the enviroment configuration above is engough. However, most inference engine only accept .pb format or .onnx format model, such as [TensorRT](https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html). |
| 125 | +The hypeprose python training library handles the whole pipelines for developing the pose estimation system, including training, evaluating and testing. Its goal is to produce a **.npz** file that contains the well-trained model weights. |
88 | 126 |
|
89 | | - Thus, one need to convert the trained model loaded with **.npz** file weight to **.pb** format or **.onnx** format for further deployment, which need extra configuration below:<br> |
| 127 | +For the training platform, the enviroment configuration above is engough. However, most inference engine accepts ProtoBuf or [ONNX](https://onnx.ai/) format model. For example, the HyperPose C++ inference engine leverages [TensorRT](https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html) as the DNN engine, which takes ONNX models as inputs. |
90 | 128 |
|
91 | | -### (I)Convert to .pb format:<br> |
92 | | - To convert the model into .pb format, we use *@tf.function* to decorate the *infer* function of each model class, so we can use the *get_concrete_function* function from tensorflow to consctruct the frozen model computation graph and then save it in .pb format. |
| 129 | +Thus, one need to convert the trained model loaded with **.npz** file weight to **.pb** format or **.onnx** format for further deployment, which need extra configuration below: |
93 | 130 |
|
94 | | - We already provide a script with cli to facilitate conversion, which located at [export_pb.py](https://github.com/tensorlayer/hyperpose/blob/master/export_pb.py). What we need here is only *tensorflow* library that we already installed. |
| 131 | +#### Converting a ProtoBuf model |
95 | 132 |
|
96 | | -### (II)Convert to .onnx format:<br> |
97 | | - To convert the model in .onnx format, we need to first convert the model into .pb format, then convert it from .pb format into .onnx format. Two extra library are needed: |
| 133 | +To convert the model into ProtoBuf format, we use `@tf.function` to decorate the `infer` function for each model class, and we then can use the `get_concrete_function` function from tensorflow to consctruct the frozen model computation graph and then save it with ProtoBuf format. |
98 | 134 |
|
99 | | -* [tf2onnx](https://github.com/onnx/tensorflow-onnx):<br> |
100 | | -*tf2onnx* is used to convert .pb format model into .onnx format model. more information see [here](https://github.com/onnx/tensorflow-onnx).<br> |
101 | | -install tf2onnx by running: |
| 135 | +We provide [a commandline tool](https://github.com/tensorlayer/hyperpose/blob/master/export_pb.py) to facilitate the conversion. The prerequisite of this tool is a tensorflow library installed along with HyperPose's dependency. |
102 | 136 |
|
103 | | - ```bash |
104 | | - pip install -U tf2onnx |
105 | | - ``` |
| 137 | +#### Converting a ONNX model |
106 | 138 |
|
107 | | -* [graph_transforms](https://github.com/tensorflow/tensorflow/tree/master/tensorflow/tools/graph_transforms#using-the-graph-transform-tool):<br> |
108 | | -*graph_transform* is used to check the input and output node of the .pb file if one doesn't know. when convert .pb file into .onnx file using tf2onnx, one is required to provide the input node name and output node name of the computation graph stored in .pb file, so he may need to use *graph_transform* to inspect the .pb file to get node names.<br> |
109 | | -build graph_transforms according to [tensorflow tools](https://github.com/tensorflow/tensorflow/tree/master/tensorflow/tools/graph_transforms#using-the-graph-transform-tool). |
| 139 | +To convert a trained model into ONNX format, we need to first convert the model into ProtoBuf format, we then convert a ProtoBuf model into ONNX format, which requires an additional library: [**tf2onnx**](https://github.com/onnx/tensorflow-onnx) for converting TensorFlow's ProtoBuf model into ONNX format. |
110 | 140 |
|
111 | | -## Extra configuration for parallel training |
112 | | - The hyperpose python training library use the High performance distributed machine learning framework **Kungfu** for parallel training.<br> |
113 | | - Thus to use the parallel training functionality of hyperpose, please install [Kungfu](https://github.com/lsds/KungFu) according to the official instructon it provides. |
| 141 | +To install `tf2onnx`, we simply run: |
114 | 142 |
|
| 143 | +```bash |
| 144 | +pip install -U tf2onnx |
| 145 | +``` |
115 | 146 |
|
| 147 | +### Extra configuration for distributed training with KungFu |
116 | 148 |
|
| 149 | +The HyperPose python training library can also perform distributed training with [Kungfu](https://github.com/lsds/KungFu). To enable parallel training, please install [Kungfu](https://github.com/lsds/KungFu) according to its official instructon. |
0 commit comments