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.

Sign up for GitHub

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

Too slow #30

Open

ox1ygen opened this issue Apr 23, 2017 · 1 comment

ox1ygen commented Apr 23, 2017

Hello! I am running this code on amazon p2.xlarge instance, and it is working much slower on GPU than on CPU.
That is the most time-consuming place:

trainer.lua
-- renormalize linear row weights
local w = layers.linear.weight
for j = 1, w:size(1) do
renorm(w[j])
end

The version of Lua I use is 5.2. CUDA is 7.5, and cuDNN is 5.1
Do you have any suggestions?

clairett commented Oct 20, 2017

I had same issue as well, can someone tell me why?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment