Python Yolo6:YOLOV5加速（失败）

Demo

安装CUDA

1.激活环境

conda activate your_env_name

2.升级工具

pip install --upgrade setuptools pip wheel

3.安装nvidia-pyindex模块

pip install nvidia-pyindex

4.安装cuda运行包

py -m pip install nvidia-cuda-runtime-cu11

5.使用conda安装cuda Toolkit组件

conda install cuda -c nvidia

这python wheels文件目前仅支持 Python 版本 3.6 到 3.9 和 CUDA 11.x，不适用于其他 Python 或 CUDA 版本。目前仅支持 Linux 操作系统和 x86\_64 CPU 架构。这些轮子文件预计可在 CentOS 7 或更高版本以及 Ubuntu 18.04 或更高版本上运行。

1.激活环境

conda activate your_env_name

2.安装tensorRT python wheel

pip install -U nvidia-tensorrt --index-url https://pypi.ngc.nvidia.com

3.验证

python3
>>> import tensorrt
>>> print(tensorrt.__version__)
>>> assert tensorrt.Builder(tensorrt.Logger())

TensorRT] ERROR: CUDA initialization failure with error 100. Please check your CUDA installation: ...

报错

win10/linux超简单安装TensorRT 在python使用_凌十一的博客-CSDN博客_linux安装tensorrt

以下安装的 仅适用于python！ c/c++不能使用 请注意，使用这种安装方法，CUDA 安装环境是通过 pip 管理的，并且必须额外注意设置您的主机环境以在 pip 环境之外使用 CUDA。 conda activate your_env_name pip install --upgrade setuptools pip wheel py -m pip 安装 nvidia-pyindex py -m pip install nvidia-cuda-runtime-cu11 conda install cuda -c nvidia 这python wheels文件目前仅支持 Python 版本 3.6 到 3.9 和 CUDA 11.x，不适用于其他 Python 或 CUDA 版本。目前仅支持 Linux 操作系统和 x86_64 CPU 架构。这些轮子文件预计可在 CentOS 7 或更高版本以及 Ubuntu 18.04 或更高版本上运行。 conda activate your_env_name pip install -U nvidia-tensorrt --index-url https://pypi.ngc.nvidia.com python3 >>> import tensorrt >>> print(tensorrt.__version__) >>> assert tensorrt.Builder(tensorrt.Logger()) 如果显示如下消息，可能没有更新英伟达驱动，更新驱动即可。 TensorRT] ERROR: CUDA initialization failure with error 100.

https://blog.csdn.net/weixin_43356770/article/details/124224193?csdn_share_tail=%7B%22type%22%3A%22blog%22%2C%22rType%22%3A%22article%22%2C%22rId%22%3A%22124224193%22%2C%22source%22%3A%22weixin_43356770%22%7D&ctrtid=mdBUu