返回顶部
首页 > 资讯 > 后端开发 > Python >ModuleNotFoundError: No module named ‘cuda‘、‘tensorrt‘
  • 307
分享到

ModuleNotFoundError: No module named ‘cuda‘、‘tensorrt‘

pythonlinux 2023-09-04 21:09:30 307人浏览 八月长安

Python 官方文档:入门教程 => 点击学习

摘要

1、 ModuleNotFoundError: No module named ‘cuda’ python -m pip install --upgrade pippip install cuda-P

1、 ModuleNotFoundError: No module named ‘cuda’

python -m pip install --upgrade pippip install cuda-Python

2、 ModuleNotFoundError: No module named ‘tensorrt’

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
在这里插入图片描述

点我进入官方TensorRT组件的版本管理
在这里插入图片描述

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的官方在线文档链接

点我进入TensorRT的官方在线文档链接
在这里插入图片描述

来源地址:https://blog.csdn.net/qq_37700257/article/details/130187061

--结束END--

本文标题: ModuleNotFoundError: No module named ‘cuda‘、‘tensorrt‘

本文链接: https://www.lsjlt.com/news/394394.html(转载时请注明来源链接)

有问题或投稿请发送至: 邮箱/279061341@qq.com    QQ/279061341

猜你喜欢
软考高级职称资格查询
编程网,编程工程师的家园,是目前国内优秀的开源技术社区之一,形成了由开源软件库、代码分享、资讯、协作翻译、讨论区和博客等几大频道内容,为IT开发者提供了一个发现、使用、并交流开源技术的平台。
  • 官方手机版

  • 微信公众号

  • 商务合作