-
Notifications
You must be signed in to change notification settings - Fork 46
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
CUDNN=1 is not working ? #23
Comments
Which version of CUDA8.0 is this ? |
@prabindh |
I strongly feel it may not be related to CUDNN. Did you stop the training in both of them after reasonable accuracies have been obtained in training ? Can you let the CUDNN version run longer epochs and check ? |
@kidapu have you sorted this? I have the same problem. I trained tiny yolo and it only works when CUDNN = 0. But this problem only happens when I try to link libdarknet-cpp-shared.so to my program. The ./darknet binary still works fine. My environment: |
@bobeo |
@bobeo Have you ensured your wrapper application (that uses the .so) also has the same options that are used for building the darknet shared lib ? |
@kidapu Does inference work with CUDNN=1, with the shared lib ? |
In summary, the following happens in my case. (1) CuDNN == 0 && ( darknet-cpp || darknet-cpp-shared) (2) CuDNN == 1 && ( darknet-cpp || darknet-cpp-shared)
|
Is this behaviour seen with the latest master as well ? Please check the latest master and confirm |
I need to confirm but I have this behaviour on v6.5-1-g372b25d with a GPU machine:
|
@ooobelix please confirm - that you are building Arapaho, and darknet with same options (for GPU, CUDNN) in both the Makefiles. |
I'm working on!
After that, I'm using my own code with Arapaho to do some predictions. Thanks for your help! |
Could you confirm, what cfg is being used ? |
From GIT:
With 0.1 threshold |
I'm:
Results:
|
I think you already tried with GPU=1, but I observed that in the last comment GPU is not defined.
|
Sorry it's a mistake, you are right! I have already tested with GPU=1 and CUDNN=1 |
I tried the Arapaho build (Windows build from darknet-cpp-windows) with latest code, and the config:- Yolo-tinyv3 cfg, and CUDA91. I am able to see detections with the default yolov3 weights. |
Ok, I did a stupid mistake into CMakeFile with the GPU and CUDNN options. Now it works well with GPU=1 and CUDNN=1 but no need of linking "-lcudnn", is it normal? |
"-lcudnn" should be required. Can we close this as the issue is resolved ? |
I'm using CMakeList and "CUDNN=1" to "set(LNK_DEP [...] cudnn" and it works well. |
I trained face with FDDB Datasets ( I wrote in #13 ), and I tried to detect face, but I can't with CUDNN=1.
On the other hand, I can detect face successfully with CUDNN=0.
My Enviroment is below.
The text was updated successfully, but these errors were encountered: