Skip to content
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 7 #10

Closed
bafonso opened this issue Nov 28, 2018 · 5 comments
Closed

cudnn 7 #10

bafonso opened this issue Nov 28, 2018 · 5 comments

Comments

@bafonso
Copy link

bafonso commented Nov 28, 2018

Is it possible to recompile this for cudnn 7? I could attempt to do it if I could get some pointers.

@bafonso bafonso closed this as completed Dec 4, 2018
@zeruniverse
Copy link
Owner

Can you install distro? I believe this comes with all setups for torch.

I will not suggest to compile CUDNN or CUDA. And I do believe there should be torch release supporting the newest CUDA or CUDNN even though it’s no longer under development.

@bafonso
Copy link
Author

bafonso commented Dec 4, 2018

The problem is that the default cudnn bindings are for cudnn5. But you can get the R7 branch and then it compiles fine for cudnn 7. I'm not finding it much faster than my cpu, is that normal? :)

@zeruniverse
Copy link
Owner

Oh, I was about to tell you this: torch/torch7#1133 (comment)

So generating optical flow is pretty slow and it's on CPU. but for applying styles you should see improvements on GPU. (run nvidia-smi to see if GPU is actually used)

@bafonso
Copy link
Author

bafonso commented Dec 5, 2018

Thanks. Is there a way to compute optical flow on the GPU? Would it be faster? :)

@zeruniverse
Copy link
Owner

I actually tried OpenACC for deepflow and it does not work well. The reason is there're very limited parallelizations you can do given your image is not so large and the optimization steps shoud be run epoch by epoch which are serial. I would suggest using OpenMP to parallelize the optical flow algorithm if it hasn't been multi-threading.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants