环境
Ubuntu18 64位 Conda3.7 Cuda Cudnn Tensorrt
安装过程参照2021-06-06 UBUNTU18 TENSORRT安装
安装
git clone https://github.com/NVIDIA-AI-IOT/torch2trt
cd torch2trt
python setup.py install
# 加上pulgin会失败
pip install tqdm cython pycocotools
sudo apt-get install python3-matplotlib
git clone https://github.com/NVIDIA-AI-IOT/trt_pose
cd trt_pose
python setup.py install
# 安装jupyter lab
pip install jupyterlab
cd tasks/humanpose
jupyter lab
# 然后可以运行demo了