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?
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:
The version of Lua I use is 5.2. CUDA is 7.5, and cuDNN is 5.1
Do you have any suggestions?