Add more informative message for CUDA error
This commit is contained in:
parent
5bfe504de1
commit
958dcfaeb8
1 changed files with 1 additions and 1 deletions
|
@ -15,7 +15,7 @@ if w2nn then
|
||||||
else
|
else
|
||||||
local state, ret = pcall(load_cunn)
|
local state, ret = pcall(load_cunn)
|
||||||
if not state then
|
if not state then
|
||||||
error("Failed to load CUDA modules. Please check the CUDA Settings.")
|
error("Failed to load CUDA modules. Please check the CUDA Settings.\n---\n" .. ret)
|
||||||
end
|
end
|
||||||
pcall(load_cudnn)
|
pcall(load_cudnn)
|
||||||
w2nn = {}
|
w2nn = {}
|
||||||
|
|
Loading…
Reference in a new issue