Skip to content

Error in rebuilding Warp_Network #72

@RogueWarrior34

Description

@RogueWarrior34

GPU: Nvidia RTX GeForce 5070 ti Laptop GPU
Cuda: 13.0
Torch version: 2.10.0.dev20251106+cu130
TensorRT version: 10.14.1.48
=>print(torch.cuda.get_arch_list())
['sm_75', 'sm_80', 'sm_86', 'sm_90', 'sm_100', 'sm_120']

The plugin exists at the location, also I have installed the compatible versions of cuda, cudnn, tensrorrt and torch (also added correctly in PATH).
Any way of fixing this?

==================== warp_network start ====================
Traceback (most recent call last):
File "D:\ditto\scripts\cvt_onnx_to_trt.py", line 168, in
main(onnx_dir, trt_dir, grid_sample_plugin_file=plugin_file)
File "D:\ditto\scripts\cvt_onnx_to_trt.py", line 146, in main
onnx_to_trt_for_gridsample(onnx_file, trt_file, fp16, plugin_file=plugin_file)
File "D:\ditto\scripts\cvt_onnx_to_trt.py", line 51, in onnx_to_trt_for_gridsample
load_plugin(plugin_file) # Load the custom plugin
File "D:\ditto\scripts\cvt_onnx_to_trt.py", line 16, in load_plugin
ctypes.CDLL(plugin_file, mode=ctypes.RTLD_GLOBAL, winmode=0)
File "D:\anaconda3\envs\ditto\lib\ctypes_init_.py", line 374, in init
self._handle = _dlopen(self._name, mode)
FileNotFoundError: Could not find module 'D:\ditto\checkpoints\ditto_onnx\grid_sample_3d_plug
in.dll' (or one of its dependencies). Try using the full path with constructor syntax.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions