Open LaFeuilleMorte opened 7 months ago
2024-03-05 17:28:53,505 - INFO - Initializing model finished in 16985.25ms. 2024-03-05 17:28:53,505 - INFO - Processing images ... 2024-03-05 17:28:54,974 - INFO - Processing images finished in 1468.80ms. 2024-03-05 17:28:54,974 - INFO - Running image 1/1 ... 2024-03-05 17:28:54,974 - INFO - Running model ... Segmentation fault (core dumped)
Hi! Are you using CPU for inference?
No, I print the device which the model will be sent, it's GPU. My computer is RTX 3090 + torch 2.2.1 + cuda 11.8
2024-03-05 17:28:53,505 - INFO - Initializing model finished in 16985.25ms. 2024-03-05 17:28:53,505 - INFO - Processing images ... 2024-03-05 17:28:54,974 - INFO - Processing images finished in 1468.80ms. 2024-03-05 17:28:54,974 - INFO - Running image 1/1 ... 2024-03-05 17:28:54,974 - INFO - Running model ... Segmentation fault (core dumped)