Add cloudpickle wrapper for compatibility

Author: ziyiwu9494

setup.py

@@ -13,7 +13,7 @@
     'akro>=0.0.8',
     'click>=2.0',
     # Older versions don't work with torch.save
-    'cloudpickle>=1.6.0',
+    'cloudpickle',
     'cma==2.7.0',
     'dowel==0.0.3',
     'numpy>=1.14.5',

src/garage/experiment/snapshotter.py

@@ -136,8 +136,12 @@ def save_itr_params(self, itr, params):
 
         if file_name:
             if torch:
+
+                class _pickle_module:
+                    Pickler = cloudpickle.CloudPickler
+
                 params['global_device'] = global_device()
-                torch.save(params, file_name, pickle_module=cloudpickle)
+                torch.save(params, file_name, pickle_module=_pickle_module)
             else:
                 with open(file_name, 'wb') as file:
                     cloudpickle.dump(params, file)