使用python setup.py install后import包出现ModuleNotFoundError

直接使用python setup.py install为本地安装，安装的路径没有添加到环境变量中，需要执行
export PYTHONPATH=$PYTHONPATH:刚刚安装的包路径
然后source ~/.bashrc

以安装torch2trt为例，使用源码安装

git clone https://github.com/NVIDIA-AI-IOT/torch2trt
cd torch2trt
python setup.py install

如果使用torch2trt

import torch2trt

报错

ModuleNotFoundError: No module named 'torch2trt'

解决方案为找到安装包的位置

locate torch2trt

或者使用python setup.py install时查看安装路径

最后添加到环境变量中

export PYTHONPATH=$PYTHONPATH:/usr/lib/python3.7/site-packages/torch2trt-0.4.0-py3.7.egg
source ~/.bashrc