diff --git a/docker-compose.yaml b/docker-compose.yaml index a21dae0f..f44660de 100644 --- a/docker-compose.yaml +++ b/docker-compose.yaml @@ -122,8 +122,13 @@ services: capabilities: [ gpu ] worker-ray: - build: ./morpheus-worker/ray - command: ray start --head --dashboard-host 0.0.0.0 --block + build: + context: ./morpheus-worker/ray + # if you are using a mac, uncomment this arg + # args: + # RAY_IMAGE_TAG: latest-gpu-aarch64 + image: morpheus-worker:latest + command: ray start --head --num-gpus 1 --dashboard-host 0.0.0.0 --block ports: - "8000:8000" - "8265:8265" @@ -134,7 +139,7 @@ services: condition: service_started worker-ray-deployer: - build: ./morpheus-worker/ray + image: morpheus-worker:latest command: bash -c "sleep 5 && ray job submit -- serve deploy models.yaml" environment: - RAY_ADDRESS=http://worker-ray:8265 diff --git a/morpheus-worker/ray/Dockerfile b/morpheus-worker/ray/Dockerfile index c15d91d2..5e198974 100644 --- a/morpheus-worker/ray/Dockerfile +++ b/morpheus-worker/ray/Dockerfile @@ -1,4 +1,5 @@ -FROM rayproject/ray:latest-gpu +ARG RAY_IMAGE_TAG="latest-gpu" +FROM rayproject/ray:$RAY_IMAGE_TAG # Configuration defaults ENV VENV_NAME ".venv-docker"