diff --git a/lib/layers.py b/lib/layers.py
index 7e1777a..729efa2 100644
--- a/lib/layers.py
+++ b/lib/layers.py
@@ -3,8 +3,8 @@ import sys
 import numpy as np
 import theano
 import theano.tensor as T
-from theano.sandbox.cuda import dnn
-from theano.sandbox.cuda.basic_ops import gpu_contiguous
+from theano.gpuarray import dnn
+from theano.gpuarray.basic_ops import gpu_contiguous
 from pylearn2.sandbox.cuda_convnet.filter_acts import FilterActs
 from pylearn2.sandbox.cuda_convnet.pool import MaxPool
 from pylearn2.expr.normalize import CrossChannelNormalization
diff --git a/spec_1gpu.yaml b/spec_1gpu.yaml
index 8297ec7..b61f043 100644
--- a/spec_1gpu.yaml
+++ b/spec_1gpu.yaml
@@ -1,6 +1,6 @@
 # If want to input None, use !!null

-gpu: 'gpu0'
+gpu: 'cuda0'
 sock_data: 5001

 batch_size: 128 # def: 256
diff --git a/train.py b/train.py
index 740b789..4e08ff5 100644
--- a/train.py
+++ b/train.py
@@ -38,8 +38,8 @@ def train_net(config):
         load_send_queue = None
         load_recv_queue = None

-    import theano.sandbox.cuda
-    theano.sandbox.cuda.use(config['gpu'])
+    import theano.gpuarray
+    theano.gpuarray.use(config['gpu'])
     #theano.gof.compilelock.set_lock_status(False) # Line added to disable compilelock
     import theano
     theano.config.on_unused_input = 'warn'
diff --git a/validate_performance.py b/validate_performance.py
index 4446d42..24a5d6d 100644
--- a/validate_performance.py
+++ b/validate_performance.py
@@ -34,8 +34,8 @@ def validate_performance(config):
         load_send_queue = None
         load_recv_queue = None

-    import theano.sandbox.cuda
-    theano.sandbox.cuda.use(config['gpu'])
+    import theano.gpuarray
+    theano.gpuarray.use(config['gpu'])
     import theano
     theano.config.on_unused_input = 'warn'