torch2trt_dynamic icon indicating copy to clipboard operation
torch2trt_dynamic copied to clipboard

Published 20 hours ago • verified publisher icon grimoire

install error

Open zkloveai opened this issue 4 years ago • 2 comments

(zk_mmdet2.4) yyf@zyzn:~/software/torch2trt$ sudo python setup.py install [sudo] password for yyf: Traceback (most recent call last): File "setup.py", line 8, in from build import build File "/home/yyf/software/torch2trt/build.py", line 29, in torch_dir=imp.find_module('torch')[1], ImportError: No module named torch

I install pytorch already.why ?

zkloveai avatar Sep 30 '20 05:09 zkloveai

I guess sudo here maybe cause this problem?

zhiqwang avatar Sep 30 '20 06:09 zhiqwang

you should change the build.py in line 28 as torch_dir= 'XX/python3.8/site-packages/torch',,it really works

hxyucas avatar Aug 09 '22 14:08 hxyucas