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

摸鱼肥仔

于 2023-06-13 16:27:28 发布

阅读量1.5k

点赞数 2

文章标签： python 开发语言 linux

本文链接：https://blog.csdn.net/m0_46336568/article/details/131190820

版权

当使用`pythonsetup.pyinstall`在本地安装torch2trt后，由于安装路径未添加到环境变量PYTHONPATH中，导致导入模块时出错。解决方法是找到安装包位置，将路径添加到PYTHONPATH，例如`exportPYTHONPATH=$PYTHONPATH:/usr/lib/python3.7/site-packages/torch2trt-0.4.0-py3.7.egg`，然后运行`source~/.bashrc`使更改生效。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

直接使用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