Git Product home page Git Product logo

Comments (5)

Yutong-gannis avatar Yutong-gannis commented on July 17, 2024

@lihuang218 FileNotFoundError: Could not find: cublas64_10.dll. Is it on your PATH? 找一下cuda安装目录下有没有cublas64_10.dll这个文件,或者有没有cublas64_xx.dll,如果有改成cublas64_10.dll试试。

from etsauto.

lihuang218 avatar lihuang218 commented on July 17, 2024

有一个cublas64_11.dll,修改后依旧报错

PS D:\ETSAuto\ETSAuto-1.1> python ./tools/export.py -o ./engines/yolov6s_bdd_60.onnx -e ./engines/yolov6s_bdd_60.engine --end2end
Traceback (most recent call last):
File "./tools/export.py", line 7, in
import tensorrt as trt
File "E:\python\lib\site-packages\tensorrt_init_.py", line 129, in
ctypes.CDLL(find_lib(lib))
File "E:\python\lib\ctypes_init_.py", line 373, in init
self._handle = _dlopen(self._name, mode)
FileNotFoundError: Could not find module 'C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\bin\cublas64_10.dll' (or one of its dependencies). Try using the full path with constructor syntax.
PS D:\ETSAuto\ETSAuto-1.1>

from etsauto.

Yutong-gannis avatar Yutong-gannis commented on July 17, 2024

@lihuang218 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\bin在这个目录下吗

from etsauto.

lihuang218 avatar lihuang218 commented on July 17, 2024

在我C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\bin在这个目录下。

将cuda10.2中的cublas64_10.dll复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\bin中,可以成功构建 YOLOV6 的 TensorRT 文件。

但是在构建 环境感知 的 TensorRT 文件时报错如下:
[04/14/2023-20:25:03] [TRT] [V] --------------- Timing Runner: Conv_9 + Add_10 (CudnnConvolution)
[04/14/2023-20:25:03] [TRT] [V] CudnnConvolution has no valid tactics for this config, skipping
[04/14/2023-20:25:03] [TRT] [V] --------------- Timing Runner: Conv_9 + Add_10 (CublasConvolution)
[04/14/2023-20:25:03] [TRT] [V] CublasConvolution has no valid tactics for this config, skipping
[04/14/2023-20:25:03] [TRT] [V] --------------- Timing Runner: Conv_9 + Add_10 (CaskConvolution)
[04/14/2023-20:25:03] [TRT] [V] CaskConvolution has no valid tactics for this config, skipping
[04/14/2023-20:25:03] [TRT] [V] Deleting timing cache: 435 entries, served 806 hits since creation.
[04/14/2023-20:25:03] [TRT] [E] 10: [optimizer.cpp::nvinfer1::builder::cgraph::LeafCNode::computeCosts::3626] Error Code 10: Internal Error (Could not find any implementation for node Conv_9 + Add_10.)
Traceback (most recent call last):
File "./tools/onnx2trt.py", line 55, in
build_engine(ONNX_SIM_MODEL_PATH, TENSORRT_ENGINE_PATH_PY)
File "./tools/onnx2trt.py", line 49, in build_engine
f.write(engine.serialize())
AttributeError: 'NoneType' object has no attribute 'serialize'

from etsauto.

Yutong-gannis avatar Yutong-gannis commented on July 17, 2024

@lihuang218 用trtexec转试试

from etsauto.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.