Merge pull request #30 from Denys88/vm/ray_and_installation_updates

Fixed Ray 1.0 API. Updated requirements.

requirements.txt

@@ -4,16 +4,12 @@ ray==1.0.0
 pyyaml
 setproctitle
 psutil
-torch>=1.5
+torch>=1.6
 tensorboard==1.14.0
 tensorboardX==1.6
-tensorflow-gpu==1.15.4
-tensorflow-probability==0.7.0
 opencv-python>=4.1.0.25
 gym-super-mario-bros==7.1.6
 pybullet>=2.5.0
 smac
 dm_control
 dm2gym
-
-

rl_games/tf14_runner.py

@@ -75,7 +75,7 @@ def get_prebuilt_config(self):
 
     def run_train(self):
         print('Started to train')
-        ray.init(redis_max_memory=1024*1024*1000, object_store_memory=1024*1024*1000)
+        ray.init(object_store_memory=1024*1024*1000)
         obs_space, action_space = env_configurations.get_obs_and_action_spaces_from_config(self.config)
         print('obs_space:', obs_space)
         print('action_space:', action_space)

rl_games/torch_runner.py

@@ -76,7 +76,7 @@ def get_prebuilt_config(self):
 
     def run_train(self):
         print('Started to train')
-        ray.init(redis_max_memory=1024*1024*1000, object_store_memory=1024*1024*1000)
+        ray.init(object_store_memory=1024*1024*1000)
         if self.exp_config:
             self.experiment =  experiment.Experiment(self.default_config, self.exp_config)
             exp_num = 0

setup.py

@@ -17,7 +17,7 @@
             # this setup is only for pytorch
             # 
             'gym>=0.17.2',
-            'torch>=1.5',
+            'torch>=1.6',
             'numpy>=1.16.0',
             'ray==1.0.0',
             'tensorboard>=1.14.0',

test_a2c_continuous.ipynb

@@ -130,7 +130,7 @@
    ],
    "source": [
     "import ray\n",
-    "ray.init(redis_max_memory=1024*1024*100, object_store_memory=1024*1024*100)"
+    "ray.init(object_store_memory=1024*1024*100)"
    ]
   },
   {
@@ -1048,4 +1048,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
+}