在这个网址找到适合自己的版本,尤其是找到合适的cuda版本:
https://pypi.nvidia.com/
然后直接pip 安装:
比如:
pip install https://pypi.nvidia.com/tensorrt-cu11/tensorrt-cu11-10.1.0.tar.gz
也可以:
wget https://pypi.nvidia.com/tensorrt-cu11/tensorrt-cu11-10.1.0.tar.gz
pip install tensorrt-cu11-10.1.0.tar.gz