fix: 修复 Python TensorRT 资源释放
- 移除不存在的 .destroy() 方法调用 - Python TensorRT 由 GC 管理,= None 即释放
This commit is contained in:
@@ -331,19 +331,18 @@ class TensorRTEngine:
|
|||||||
return {"total_mb": 0, "used_mb": 0, "free_mb": 0}
|
return {"total_mb": 0, "used_mb": 0, "free_mb": 0}
|
||||||
|
|
||||||
def _release_resources(self):
|
def _release_resources(self):
|
||||||
"""释放资源"""
|
"""释放资源(Python TensorRT 由 GC 管理,无需 destroy)"""
|
||||||
if self._stream:
|
if self._stream:
|
||||||
try:
|
try:
|
||||||
self._stream.synchronize()
|
self._stream.synchronize()
|
||||||
except Exception:
|
except Exception:
|
||||||
pass
|
pass
|
||||||
|
self._stream = None
|
||||||
|
|
||||||
if self._context:
|
if self._context:
|
||||||
self._context.destroy()
|
|
||||||
self._context = None
|
self._context = None
|
||||||
|
|
||||||
if self._engine:
|
if self._engine:
|
||||||
self._engine.destroy()
|
|
||||||
self._engine = None
|
self._engine = None
|
||||||
|
|
||||||
self._memory_pool.clear()
|
self._memory_pool.clear()
|
||||||
|
|||||||
Reference in New Issue
Block a user