1
0
Fork 0
mirror of synced 2024-06-28 11:10:25 +12:00

Add error message when the CUDA module loading fails

This commit is contained in:
nagadomi 2016-05-06 02:20:54 +09:00
parent da03209d3e
commit 5bfe504de1

View file

@ -13,7 +13,10 @@ end
if w2nn then
return w2nn
else
pcall(load_cunn)
local state, ret = pcall(load_cunn)
if not state then
error("Failed to load CUDA modules. Please check the CUDA Settings.")
end
pcall(load_cudnn)
w2nn = {}
require 'LeakyReLU'