Add error message when the CUDA module loading fails
This commit is contained in:
parent
da03209d3e
commit
5bfe504de1
|
@ -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'
|
||||
|
|
Loading…
Reference in a new issue