Do you have multiple GPUs on your system? You can try to set CUDA_VISIBLE_DEVICES=gpu_id_which_you_want_to_use ./your_program
This is usually related to other elements in the pipeline. Can you share your pipeline?
Do you have multiple GPUs on your system? You can try to set CUDA_VISIBLE_DEVICES=gpu_id_which_you_want_to_use ./your_program
This is usually related to other elements in the pipeline. Can you share your pipeline?