bump ray version

kingoflolz · Jul 19, 2021 · e2c99b4 · e2c99b4
1 parent 3fcb682
commit e2c99b4
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/ray_tpu.py b/ray_tpu.py
@@ -162,4 +162,4 @@ def start_ray(conn, address, version=1):
 
     time.sleep(1)
 
-    conn.run(f"TCMALLOC_LARGE_ALLOC_REPORT_THRESHOLD={32 * 1024**3} ray start --address={address} --resources='" + '{"tpu": 1}\'', hide=True)
+    conn.run(f"TCMALLOC_LARGE_ALLOC_REPORT_THRESHOLD={32 * 1024**3} ray start --address={address} --resources='" + '{"tpu": 1}\' --include-dashboard False', hide=True)
diff --git a/requirements.txt b/requirements.txt
@@ -9,7 +9,7 @@ fabric~=2.6.0
 optax==0.0.6
 git+https://github.com/deepmind/dm-haiku
 git+https://github.com/EleutherAI/lm-evaluation-harness/
-ray~=1.4.1
+ray==1.4.1
 jax~=0.2.12
 Flask~=1.1.2
 cloudpickle~=1.3.0

diff --git a/scripts/init_ray.sh b/scripts/init_ray.sh
@@ -7,4 +7,4 @@ sudo docker rm libtpu || true
 sudo docker create --name libtpu gcr.io/cloud-tpu-v2-images/libtpu:libtpu_20210518_RC00 "/bin/bash" && sudo docker cp libtpu:libtpu.so /lib
 
 # initializes jax and installs ray on cloud TPUs
-sudo pip install --upgrade jaxlib jax==0.2.12 ray==1.2.0 fabric dataclasses optax==0.0.6 git+https://github.com/deepmind/dm-haiku tqdm cloudpickle smart_open[gcs] einops func_timeout
+sudo pip install --upgrade jaxlib==0.1.68 jax==0.2.12 ray==1.4.1 fabric dataclasses optax==0.0.6 git+https://github.com/deepmind/dm-haiku tqdm cloudpickle smart_open[gcs] einops func_timeout
diff --git a/scripts/init_ray_v2.sh b/scripts/init_ray_v2.sh
@@ -3,4 +3,4 @@ set -e
 
 # initializes jax and installs ray on cloud TPUs
 sudo pip install "jax[tpu]==0.2.14" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
-sudo pip install --upgrade ray==1.2.0 fabric dataclasses optax==0.0.6 git+https://github.com/deepmind/dm-haiku tqdm cloudpickle smart_open[gcs] einops func_timeout
+sudo pip install --upgrade ray==1.4.1 fabric dataclasses optax==0.0.6 git+https://github.com/deepmind/dm-haiku tqdm cloudpickle smart_open[gcs] einops func_timeout
Original file line number	Diff line number	Diff line change
Expand Up		@@ -162,4 +162,4 @@ def start_ray(conn, address, version=1):

		time.sleep(1)

		conn.run(f"TCMALLOC_LARGE_ALLOC_REPORT_THRESHOLD={32 * 1024**3} ray start --address={address} --resources='" + '{"tpu": 1}\'', hide=True)
		conn.run(f"TCMALLOC_LARGE_ALLOC_REPORT_THRESHOLD={32 * 1024**3} ray start --address={address} --resources='" + '{"tpu": 1}\' --include-dashboard False', hide=True)