-
Notifications
You must be signed in to change notification settings - Fork 131
Description
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.