stress tests done

This commit is contained in:
Alex
2021-01-29 21:24:07 +00:00
parent b4d87b8fc5
commit 9fd198635f
3 changed files with 6 additions and 3 deletions
@@ -6,7 +6,7 @@ max_workers: 3
idle_timeout_minutes: 15
docker:
image: anyscale/ray-ml:latest-gpu
image: rayproject/ray-ml:latest-gpu
container_name: ray_container
pull_before_run: True
@@ -21,14 +21,16 @@ auth:
head_node:
InstanceType: g3.8xlarge
ImageId: ami-0828a1066dc750737
worker_nodes:
InstanceType: g3.8xlarge
ImageId: ami-0828a1066dc750737
InstanceMarketOptions:
MarketType: spot
setup_commands:
- apt-get install -y libglib2.0-0 libcudnn7=7.6.5.32-1+cuda10.1
- sudo apt-get install -y libglib2.0-0 libcudnn7=7.6.5.32-1+cuda10.1
- pip install -U https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-2.0.0.dev0-cp37-cp37m-manylinux2014_x86_64.whl
# Command to start ray on the head node. You don't need to change this.
@@ -43,6 +43,7 @@ echo "branch: $ray_branch"
echo "workload: $workload"
wheel="https://s3-us-west-2.amazonaws.com/ray-wheels/$ray_branch/$commit/ray-$ray_version-cp37-cp37m-manylinux2014_x86_64.whl"
wheel="https://ray-wheels.s3-us-west-2.amazonaws.com/releases/1.2.0/b4d87b8fc5ff03b868adba35b37da32b7be9f71f/ray-1.2.0-cp37-cp37m-manylinux2014_x86_64.whl"
conda uninstall -y terminado || true
pip install -U pip
+1 -1
View File
@@ -91,7 +91,7 @@ setup_commands:
- pip install terminado
- pip install boto3==1.4.8 cython==0.29.0
# - cd ray/python; git checkout master; git pull; pip install -e . --verbose
- pip install -U pip install https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-2.0.0.dev0-cp38-cp38-manylinux2014_x86_64.whl
- pip install -U pip install https://ray-wheels.s3-us-west-2.amazonaws.com/releases/1.2.0/b4d87b8fc5ff03b868adba35b37da32b7be9f71f/ray-1.2.0-cp37-cp37m-manylinux2014_x86_64.whl
# Custom commands that will be run on the head node after common setup.
head_setup_commands: []