ModuleNotFoundError: No module named ‘cuda‘、‘tensorrt‘
短信预约 -IT技能 免费直播动态提醒
python -m pip install --upgrade pippip install cuda-python
2.1 依赖库
先安装两个TensorRT的依赖库
python -m pip install --upgrade pippip install nvidia-pyindexpip install onnx-graphsurgeon
2.2 下载TensorRT 及CUDA版本环境对应
点我进入TensorRT官方下载地址, win下载后得到zip文件TensorRT-8.6.0.12.Windows10.x86_64.cuda-11.8.zip
2.3 安装TensorRT
确认本机是否成功安装CUDA Toolkit, win+R打开"运行",输入 cmd 终端。
输入指令:
# 查看cuda版本nvcc -V# 查看Python版本 ---作者是Python 3.8.15python -V
找到本机安装CUDA的文件路径:
- 默认安装通常在:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3(版本)
- 本文安装主路径在:
D:\ProgramFiles\NVIDIAGPUComputingToolkit\CUDA\v11.3
- 解压TensorRT后主路径:
D:\TensorRT-8.6.0.12
随后将如下TensorRT中的部分文件移动到CUDA主路径下的对应文件夹内。(* 代表该目录下所有文件放入到目标路径, *.lib 表示当前文件夹下所有.lib结尾的文件)
:
TensorRT 文件夹下 | CUDA 文件夹下 |
---|---|
./bin/* | ./bin/ |
./include/* | ./include/ |
./lib/*.lib | ./lib/x64/ |
./lib/*.dll | ./bin/ |
win+R打开"运行",输入 cmd 终端,输入如下指令(作者的Python是 3.8.15):
cd d:cd D:\TensorRT-8.6.0.12\# 激活conda环境, 或者本机的Python环境pip install .\graphsurgeon\graphsurgeon-0.4.6-py2.py3-none-any.whlpip install .\onnx_graphsurgeon\onnx_graphsurgeon-0.3.12-py2.py3-none-any.whl# 此行根据自己的Python版本选择!!!pip install .\python\tensorrt-8.6.0-cp38-none-win_amd64.whlpip install .\uff\uff-0.6.9-py2.py3-none-any.whl
输入如下代码验证TensorRT是否成功安装:
import tensorrt as trtprint(trt.__version__)
最后老规矩,附上成功安装TensorRT 8的截图。
2.4 附上学习TensorRT的官方在线文档链接
来源地址:https://blog.csdn.net/qq_37700257/article/details/130187061
免责声明:
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341