# Install Ray using pip
!pip install ray
import ray
import time
import os
ray.__version__
# A toy example to demonstrate normal function in python which might let's say 2 seconds to complete
def basic_function(x):
time.sleep(2)
return x*x
# Exact replicate of the previous function but with a decorator of ray
@ray.remote
def ray_function(x):
time.sleep(2)
return x*x
%%time
# Run the basic python function 4 times
y = [basic_function(x) for x in range(4)]
y
# Init function of ray the resources can be specified here, for example num_cpus, num_gpus etc.
ray.init(num_cpus=os.cpu_count())
%%time
# Ray function can be used with same arguments but need remote() is added
futures = [ray_function.remote(x) for x in range(4)]
%%time
# Get function takes all the list of futures from the previous output and materalize the results
y_h = ray.get(futures)
y_h
# Check if ray output y_h and python outout y are the same indeed
assert y_h == y
