diff --git a/dnc/util.py b/dnc/util.py
index 87ae5d6..e16e0be 100644
--- a/dnc/util.py
+++ b/dnc/util.py
@@ -29,7 +29,7 @@ def cuda(x, grad=False, gpu_id=-1):
     t.requires_grad=grad
     return t
   else:
-    t = T.FloatTensor(x.pin_memory()).cuda(gpu_id, async=True)
+    t = T.FloatTensor(x.pin_memory()).cuda(gpu_id)
     t.requires_grad=grad
     return t
 
@@ -40,7 +40,7 @@ def cudavec(x, grad=False, gpu_id=-1):
     t.requires_grad = grad
     return t
   else:
-    t = T.Tensor(T.from_numpy(x).pin_memory()).cuda(gpu_id, async=True)
+    t = T.Tensor(T.from_numpy(x).pin_memory()).cuda(gpu_id)
     t.requires_grad = grad
     return t
 
@@ -51,7 +51,7 @@ def cudalong(x, grad=False, gpu_id=-1):
     t.requires_grad = grad
     return t
   else:
-    t = T.LongTensor(T.from_numpy(x.astype(np.long)).pin_memory()).cuda(gpu_id, async=True)
+    t = T.LongTensor(T.from_numpy(x.astype(np.long)).pin_memory()).cuda(gpu_id)
     t.requires_grad = grad
     return t
 
diff --git a/setup.py b/setup.py
index c076a51..9e47555 100644
--- a/setup.py
+++ b/setup.py
@@ -22,7 +22,7 @@ with open(path.join(here, 'README.rst'), encoding='utf-8') as f:
 setup(
     name='dnc',
 
-    version='1.0.0',
+    version='1.0.1',
     description='Differentiable Neural Computer, for Pytorch',
     long_description=long_description,