ImportError: нет модуля с именем tenorrt

Я установил TensorRT на свою виртуальную машину с помощью установки Debian. Если я запускаю "dpkg -l | grep TensorRT", я получаю ожидаемый результат:

ii  graphsurgeon-tf                                             5.0.2-1+cuda10.0                           amd64        GraphSurgeon for TensorRT package
ii  libnvinfer-dev                                              5.0.2-1+cuda10.0                           amd64        TensorRT development libraries and headers
ii  libnvinfer-samples                                          5.0.2-1+cuda10.0                           all          TensorRT samples and documentation
ii  libnvinfer5                                                 5.0.2-1+cuda10.0                           amd64        TensorRT runtime libraries
ii  python3-libnvinfer                                          5.0.2-1+cuda10.0                           amd64        Python 3 bindings for TensorRT
ii  python3-libnvinfer-dev                                      5.0.2-1+cuda10.0                           amd64        Python 3 development package for TensorRT
ii  tensorrt                                                    5.0.2.6-1+cuda10.0                         amd64        Meta package of TensorRT
ii  uff-converter-tf                                            5.0.2-1+cuda10.0                           amd64        UFF converter for TensorRT package

Но когда я пытаюсь импортировать его в Python, я получаю:

File "<stdin>", line 1, in <module>
ImportError: No module named 'tensorrt'

Что я делаю неправильно?


python-3.x linux tensorrt
person K41F4r    schedule 07.04.2019    source источник
comment
Вы пытаетесь импортировать в Python 2 или Python 3?   -  person mibrahimy    schedule 17.04.2019
comment
Я использую Python 3   -  person K41F4r    schedule 17.04.2019
comment
Пожалуйста, попробуйте импортировать uff и tensorflow до tensorRT.   -  person mibrahimy    schedule 17.04.2019
comment
Ваша проблема решена?   -  person mibrahimy    schedule 19.04.2019
comment
У меня больше нет среды для тестирования, но в то время, когда я импортировал uff, она не была найдена   -  person K41F4r    schedule 19.04.2019
comment
Я тоже в одной лодке. Все установил, с dpkg -l | grep TensorRT, я тоже вижу установленный пакет python, но не отображается.   -  person Jagesh Maharjan    schedule 14.06.2019
comment
То же самое. Какую версию python3 вы используете?   -  person Artemis F    schedule 17.06.2019

Ответы (3)


arrow_upward
3
arrow_downward

У меня была такая же ошибка, я решил ее, установив следующие 2 пакета:

# from TensorRT Tar Package ( TensorRT-5.0.2.6/python/ ) 
pip3 install tensorrt-5.0.2.6-py2.py3-none-any.whl  
# pycuda
pip3 install pycuda 

# libnvinfer 5.0.2-1
# libnvinfer-dev 5.0.2-1
# libnvinfer-samples 5.0.2-1
# python3-libnvinfer 5.0.2-1
# python3-libnvinfer-dev 5.0.2-1
# tensorrt 5.0.2.6
person sailfish009    schedule 06.12.2019

arrow_upward
0
arrow_downward

два предложения:

  1. убедитесь, что ваш TensorRT добавлен, укажите свой путь (LD_LIBRARY_PATH)
  2. если у вас установлены как python 2, так и python3, проверьте установку пакета в списке pip и / или списке pip3
person Pidem    schedule 01.08.2019

arrow_upward
0
arrow_downward

Вам нужно импортировать в среду python3, поскольку вы, кажется, установили TensorRT для python3. Таким образом, он не импортируется на python2.

person VARAT BOHARA    schedule 29.08.2020