diff --git a/common/h36m_dataset.py b/common/h36m_dataset.py
index 3778672..459e74f 100755
--- a/common/h36m_dataset.py
+++ b/common/h36m_dataset.py
@@ -231,7 +231,7 @@ def __init__(self, path, remove_static_joints=True):
                                                    cam['tangential_distortion']))
         
         # Load serialized dataset
-        data = np.load(path)['positions_3d'].item()
+        data = np.load(path, allow_pickle=True)['positions_3d'].item()
         
         self._data = {}
         for subject, actions in data.items():
@@ -252,4 +252,4 @@ def __init__(self, path, remove_static_joints=True):
             
     def supports_semi_supervised(self):
         return True
-   
\ No newline at end of file
+   
diff --git a/run.py b/run.py
index b71e94b..9915cc3 100755
--- a/run.py
+++ b/run.py
@@ -61,7 +61,7 @@
         anim['positions_3d'] = positions_3d
 
 print('Loading 2D detections...')
-keypoints = np.load('data/data_2d_' + args.dataset + '_' + args.keypoints + '.npz')
+keypoints = np.load('data/data_2d_' + args.dataset + '_' + args.keypoints + '.npz', allow_pickle=True)
 keypoints_metadata = keypoints['metadata'].item()
 keypoints_symmetry = keypoints['metadata'].item()['keypoints_symmetry']
 kps_left, kps_right = list(keypoints_symmetry[0]), list(keypoints_symmetry[1])
@@ -204,8 +204,9 @@ def fetch(subjects, action_filter=None, subset=1, parse_3d_poses=True):
                             filter_widths=filter_widths, causal=args.causal, dropout=args.dropout, channels=args.channels,
                             dense=args.dense)
 
-model_pos_train=nn.DataParallel(model_pos_train,device_ids=[0,1,2]) # multi-GPU
-model_pos=nn.DataParallel(model_pos,device_ids=[0,1,2]) # multi-GPU
+device_list = [i for i in range(torch.cuda.device_count())]
+model_pos_train=nn.DataParallel(model_pos_train,device_ids=device_list) # multi-GPU
+model_pos=nn.DataParallel(model_pos,device_ids=device_list) # multi-GPU
 
 
 receptive_field = model_pos.module.receptive_field()