search

pgigioli / darknet_ros

YOLO integration with ROS for real-time object detection
50 stars 23 forks source link

Error with CUDA #5

Closed wang19950602 closed 7 years ago

wang19950602 commented 7 years ago

Hello, I have got a little problem when I catkin_make, error with cuda I got 2 GPU in my sever, GTX960 and Tesla K40, it seems that the darknet_ros node choose the GTX960 as default GPU, how can I change to Tesla K40. Just like in yolo, I can change gpu by adding option "-i 1". Thank you!

pgigioli commented 7 years ago

There isn't a functional way of changing the GPU device but you can manually change the gpu_index param in 'cuda.c'

wang19950602 commented 7 years ago

Much appreciate for your quick reply. Actually, the problem is not the choice of gpu device, I change the 'compute_30，sm_30' to 52 in CMakeFile.txt, which is the corresponding compute ability of GTX960, and finally it works!