Add error message when the CUDA module loading fails
This commit is contained in:
parent
da03209d3e
commit
5bfe504de1
1 changed files with 4 additions and 1 deletions
|
@ -13,7 +13,10 @@ end
|
||||||
if w2nn then
|
if w2nn then
|
||||||
return w2nn
|
return w2nn
|
||||||
else
|
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)
|
pcall(load_cudnn)
|
||||||
w2nn = {}
|
w2nn = {}
|
||||||
require 'LeakyReLU'
|
require 'LeakyReLU'
|
||||||
|
|
Loading…
Reference in a new issue