Exe question during test process.

hmmlillian / Neural-Color-Transfer

The source code of "single-reference color transfer" in "Progressive Color Transfer with Dense Semantic Correspondences".

https://arxiv.org/abs/1710.00756

74 stars 8 forks source link

Closed Ronales closed 4 years ago

Ronales commented 4 years ago

Dear author, When I execute exe file to test IMG, Should I also build caffe2 environment?

this is some error in executing process:

Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR Check failure stack trace:

I try to fix this problem by add "engine=caffe" and so on. but finally, I failed. Can you give me some solve method? Thanks in advance!

hmmlillian commented 4 years ago

@Ronales I think you can disable TDR to solve this problem. To disable TDR, you may change the TDR-related registry keys like this https://docs.microsoft.com/en-us/windows-hardware/drivers/display/tdr-registry-keys, or use NVIDIA Nsight Development Platform (https://docs.nvidia.com/gameworks/content/developertools/desktop/timeout_detection_recovery.htm) if you use NVIDIA GPU on Windows.

Ronales commented 4 years ago

Thanks for your reply！

deladeladela commented 3 months ago

Thanks for your reply！

请问问题解决了吗？