forked from akash-network/awesome-akash
-
Notifications
You must be signed in to change notification settings - Fork 0
/
ray-nightly-py310-gpu.yml
121 lines (117 loc) · 2.6 KB
/
ray-nightly-py310-gpu.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
version: '2.0'
services:
ray-head:
image: rayproject/ray:nightly-py310-gpu
expose:
- port: 8265
as: 8265
to:
- service: ray-worker
- global: true
- port: 6380
as: 6380
to:
- service: ray-worker
- global: true
- port: 8078
as: 8078
to:
- service: ray-worker,
- port: 8079
as: 8079
to:
- service: ray-worker
command:
- "ray"
- "start"
- "--head"
- "--port=6380"
- "--dashboard-port=8265"
- "--dashboard-host=0.0.0.0"
- "--object-manager-port=8076"
- "--node-manager-port=8077"
- "--dashboard-agent-grpc-port=8078"
- "--dashboard-agent-listen-port=8079"
- "--block"
env:
- RAY_ADDRESS=ray-head:6380
params:
storage:
shm:
mount: /dev/shm
ray-worker:
image: rayproject/ray:nightly-py310-gpu
expose:
- port: 8078
as: 8078
to:
- global: true
- port: 8079
as: 8079
to:
- global: true
command:
- /bin/sh
- -c
- |
ray start --address=ray-head:6380 --object-manager-port=8076 --node-manager-port=8077 --dashboard-agent-grpc-port=8078 --dashboard-agent-listen-port=8079 --min-worker-port=10002 --max-worker-port=10007 --block
params:
storage:
worker-shm:
mount: /dev/shm
profiles:
compute:
ray-head:
resources:
cpu:
units: 5
memory:
size: 10Gi
gpu:
units: 1
attributes:
vendor:
nvidia:
storage:
- size: 10Gb
- name: shm
size: 15Gi
attributes:
persistent: false
class: ram
ray-worker:
resources:
cpu:
units: 5
memory:
size: 10Gi
gpu:
units: 1
attributes:
vendor:
nvidia:
storage:
- size: 10Gb
- name: worker-shm
size: 15Gi
attributes:
persistent: false
class: ram
placement:
westcoast:
pricing:
ray-head:
denom: uakt
amount: 1000000
ray-worker:
denom: uakt
amount: 1000000
deployment:
ray-head:
westcoast:
profile: ray-head
count: 1
ray-worker:
westcoast:
profile: ray-worker
count: 1