Updated ray_autoscale and requirements.txt

flow-project · May 21, 2020 · a524f4d · a524f4d
1 parent 3154f26
commit a524f4d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 gym==0.14.0
-numpy==1.16.0
+numpy==1.18.4
 scipy==1.1.0
 lxml==4.4.1
 pyprind==2.11.2

diff --git a/scripts/ray_autoscale.yaml b/scripts/ray_autoscale.yaml
@@ -68,11 +68,20 @@ worker_nodes:
 
 setup_commands:
     - cd flow && git fetch && git checkout origin/master
-
-head_setup_commands:
+    - pip install ray==0.8.0
+    - pip install tabulate
     - pip install boto3==1.10.45  # 1.4.8 adds InstanceMarketOptions
     - pip install awscli==1.16.309
+    - pip install stable-baselines
     - pip install pytz
+    - pip install torch==1.3.1
+    - pip install tensorflow==2.0.0
+    - pip install lz4
+    - pip install dm-tree
+    - pip install numpy==1.18.4
+    - ./flow/scripts/setup_sumo_ubuntu1604.sh
+
+head_setup_commands: []
 
 # Custom commands that will be run on worker nodes after common setup.
 worker_setup_commands: []