From 56582b8401ba96ad729c22e4a5865540f2ba40cf Mon Sep 17 00:00:00 2001
From: nagadomi <nagadomi@nurs.or.jp>
Date: Wed, 9 Dec 2015 07:04:04 +0900
Subject: [PATCH] Fix error in training script due to missing libcudnn.so

---
 lib/srcnn.lua | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/srcnn.lua b/lib/srcnn.lua
index 29be477..0c11450 100644
--- a/lib/srcnn.lua
+++ b/lib/srcnn.lua
@@ -9,7 +9,7 @@ function nn.SpatialConvolutionMM:reset(stdv)
    self.weight:normal(0, stdv)
    self.bias:zero()
 end
-if cudnn then
+if cudnn and cudnn.SpatialConvolution then
    function cudnn.SpatialConvolution:reset(stdv)
       stdv = math.sqrt(2 / ((1.0 + 0.1 * 0.1) * self.kW * self.kH * self.nOutputPlane))
       self.weight:normal(0, stdv)