Explorar el Código

Add error message when the CUDA module loading fails

nagadomi hace 9 años
padre
commit
5bfe504de1
Se han modificado 1 ficheros con 4 adiciones y 1 borrados
  1. 4 1
      lib/w2nn.lua

+ 4 - 1
lib/w2nn.lua

@@ -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'