728x90
반응형
[TensorRT] ERROR: runtime.cpp (24) - Cuda Error in allocate: 2
[TensorRT] ERROR: runtime.cpp (24) - Cuda Error in allocate: 2
Cuda Memory 설정과 관련된 문제로서
builder.max_workspace_size = 1 << 30
적절하게 설정해준다.
728x90
반응형
'AI Development > TensorRT' 카테고리의 다른 글
| [TensorRT] Yolo v3 to onnx (2) | 2019.10.08 |
|---|---|
| [TensorRT] support matrix 지원 가능한 레이어 목록 (0) | 2019.09.26 |
| [TensorRT] keras model 에서 node name 확인하는 방법 (1) | 2019.09.19 |
| [TensorRT] 마지막 노드 찾기 (0) | 2019.08.03 |
| [TensorRT] TF-TRT vs TRT (0) | 2019.08.03 |