diff --git a/lib/nn/parallel/data_parallel.py b/lib/nn/parallel/data_parallel.py
index 0ffbfb6..8e9b72b 100644
--- a/lib/nn/parallel/data_parallel.py
+++ b/lib/nn/parallel/data_parallel.py
@@ -13,7 +13,7 @@ def async_copy_to(obj, dev, main_stream=None):
     if torch.is_tensor(obj):
         obj = Variable(obj)
     if isinstance(obj, Variable):
-        v = obj.cuda(dev, async=True)
+        v = obj.cuda(dev, non_blocking=True)
         if main_stream is not None:
             v.data.record_stream(main_stream)
         return v