AWS EC2 Python ModuleNotFoundError

Question

我正在使用具有基于 Ubuntu 18.04 的深度学习 AMI 的 EC2 实例。

我正在执行以下操作：

• 启动终端
• 激活 conda 环境： conda activate tensorflow2_latest_p37
• 安装package pip3 install tensorrt
• 运行我的代码python3 mycode.py

我收到以下错误：

ModuleNotFoundError: No module named 'tensorrt'

我也尝试安装为sudo或使用-U或-m选项。 没有任何工作。 我无法使用 conda install 安装，因为 tensorrt 在其中不可用。

Answer 1

pip3将使用 ubuntu 的级别 pip3，而不是来自 anaconda。 您可以在激活环境后使用which pip3来确认这一点。 tensorrt 不适用于 python 3.6 以外的版本

所以要在 Ubuntu 18.04 深度学习上安装tensorrt ：

conda activate tensorflow2_p36

# then in tensorflow2_p36 environment

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