Блог

Я использую экземпляр EC2 с AMI глубокого обучения на основе Ubuntu 18.04.

Я делаю следующее:

• Запустите терминал
• Активируйте среду conda: conda activate tensorflow2_latest_p37
• Установите пакет pip3 install tensorrt
• Запустите мой код python3 mycode.py

И я получаю следующую ошибку:

 ModuleNotFoundError: No module named 'tensorrt'
 

Я также пытаюсь установить как sudo или с опцией -U или -m. Ничего не работает. Я не могу установить с помощью conda install, потому что tensorrt в нем недоступен.

pip3 будет использоваться pip3 уровня ubuntu, а не от anaconda. Вы можете подтвердить это с помощью which pip3 после активации среды. Также tensorrt недоступен для python, кроме 3.6

Итак, чтобы установить tensorrt на Ubuntu 18.04 Глубокое обучение:

 conda activate tensorflow2_p36

# then in tensorflow2_p36 environment

pip install nvidia-pyindex
pip install --upgrade nvidia-tensorrt