Skip to content

Commit

Permalink
Fix several release tests (#37022)
Browse files Browse the repository at this point in the history
Signed-off-by: can <can@anyscale.com>
  • Loading branch information
can-anyscale authored Jul 3, 2023
1 parent 82172e8 commit ac588a8
Show file tree
Hide file tree
Showing 4 changed files with 254 additions and 22 deletions.
5 changes: 1 addition & 4 deletions release/ray_release/byod/byod_rllib_test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -6,17 +6,14 @@ set -exo pipefail

git clone https://github.com/ray-project/rl-experiments.git
unzip rl-experiments/halfcheetah-sac/2022-12-17/halfcheetah_1500_mean_reward_sac.zip -d ~/.
# Use torch+CUDA10.2 for our release tests. CUDA11.x has known performance issues in combination with torch+GPU+CNNs
# TODO(sven): remove once nightly image gets upgraded.
pip3 install torch==1.12.1+cu102 torchvision==0.13.1+cu102 --extra-index-url https://download.pytorch.org/whl/cu102
pip3 install torch==2.0.0+cu118 torchvision==0.15.1+cu118 --index-url https://download.pytorch.org/whl/cu118

# TODO(sven): remove once nightly image gets gymnasium and the other new dependencies.
wget https://mujoco.org/download/mujoco210-linux-x86_64.tar.gz
mkdir ~/.mujoco
mv mujoco210-linux-x86_64.tar.gz ~/.mujoco/.
cd ~/.mujoco
tar -xf ~/.mujoco/mujoco210-linux-x86_64.tar.gz
echo "export LD_LIBRARY_PATH=\$LD_LIBRARY_PATH:/home/ray/.mujoco/mujoco210/bin" >> /home/ray/.bashrc

# not strictly necessary, but makes debugging easier
git clone https://github.com/ray-project/ray.git
6 changes: 6 additions & 0 deletions release/ray_release/byod/requirements_ml_byod_3.8.in
Original file line number Diff line number Diff line change
Expand Up @@ -10,19 +10,25 @@ gcsfs
gsutil
gym
gymnasium[atari,mujoco]
memray
mujoco-py
modin
numpy
openskill
petastorm
pyarrow
pytest
semidbm
tblib
tensorboardX
torch
torchtext
torchvision
trueskill
tqdm
typer
urllib3
uvicorn
validators
wandb
xgboost
Loading

0 comments on commit ac588a8

Please sign in to comment.