code_public/lidar_rendu
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove broken CuPy 13.x memory pool setup

Browse Source
This commit is contained in:
Antoine Jacquin
2026-05-31 21:35:27 +02:00
parent 5a9cdddcfb
commit 17fd96fdf5
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lidar_pipeline/gpu.py
View File

@ -123,10 +123,8 @@ def _init_gpu():
_cp = _real_cupy _cp = _real_cupy
_cp_ndimage = _real_cupy_ndimage _cp_ndimage = _real_cupy_ndimage
props = _real_cupy.cuda.runtime.getDeviceProperties(0) # Memory pool management removed — CuPy 13.x uses its own allocator.
total_mem = props['totalGlobalMem'] # OOM protection handled at the application level (gpu_cleanup() calls).
pool_size = int(total_mem * 0.9)
_real_cupy.cuda.set_memory_pool(0, pool_size)
except (ImportError, Exception) as e: except (ImportError, Exception) as e:
logger.warning(f"GPU {_best_gpu_id} ({_gpu_name}) non utilisable — mode CPU: {e}") logger.warning(f"GPU {_best_gpu_id} ({_gpu_name}) non utilisable — mode CPU: {e}")