如何在python中的类中使用射线并行?

如何解决如何在python中的类中使用射线并行?

我想使用ray task 方法而不是ray actor 方法来并行化类中的方法。后者的原因似乎需要更改实例化类的方式(如here所示)。下面是一个玩具代码示例,以及错误

import numpy as np
import ray


class MyClass(object):
    
    def __init__(self):
        ray.init(num_cpus=4)
    
    @ray.remote
    def func(self,x,y):
        return x * y
    
    def my_func(self):
        a = [1,2,3]
        b = np.random.normal(0,1,10000)
        result = []
        # we wish to parallelise over the array `a`
        for sub_array in np.array_split(a,3):
            result.append(self.func.remote(sub_array,b))
        return result

mc = MyClass()
mc.my_func()
>>> TypeError: missing a required argument: 'y'

由于ray似乎并不“意识到”该类而出现错误,因此它需要一个自变量self

如果我们不使用类,则代码可以正常工作

@ray.remote
def func(x,y):
    return x * y

def my_func():
    a = [1,3,4]
    b = np.random.normal(0,10000)
    result = []
    # we wish to parallelise over the list `a`
    # split `a` and send each chunk to a different processor
    for sub_array in np.array_split(a,4):
        result.append(func.remote(sub_array,b))
    return result


res = my_func()
ray.get(res)
>>> [array([-0.41929678,-0.83227786,-2.69814232,...,-0.67379119,-0.79057845,-0.06862196]),array([-0.83859356,-1.66455572,-5.39628463,-1.34758239,-1.5811569,-0.13724391]),array([-1.25789034,-2.49683358,-8.09442695,-2.02137358,-2.37173535,-0.20586587]),array([ -1.67718712,-3.32911144,-10.79256927,-2.69516478,-3.1623138,-0.27448782])]```

我们看到输出是预期的4个数组的列表。如何使用ray使MyClass进行并行处理?

解决方法

一些提示:

  1. 通常建议您仅在python中的函数或类上使用ray.remote装饰器(而不是绑定方法)。

  2. 在函数的构造函数中调用ray.init时应非常小心,因为ray.init并不是幂等的(这意味着程序将在以下情况下失败)您实例化了MyClass的多个实例。相反,您应该确保ray.init在程序中仅运行一次。

我认为使用Ray可以达到两种效果。

您可以将func从类中移出,这样它将成为函数而不是绑定方法。请注意,在这种方法中,MyClass将被序列化,这意味着funcMyClass所做的更改将不会在函数之外的任何地方反映出来。在您的简化示例中,这似乎没有问题。这种方法最容易实现最大并行度。

@ray.remote
def func(obj,x,y):
    return x * y


class MyClass(object):
    def my_func(self):
        ...
        # we wish to parallelise over the array `a`
        for sub_array in np.array_split(a,3):
            result.append(func.remote(self,sub_array,b))
        return result

您可以考虑的另一种方法是使用async actors。在这种方法中,ray actor将通过异步处理并发,但这comes with the limitations of asyncio

@ray.remote(num_cpus=4)
class MyClass(object):
    async def func(self,y):
        return x * y
    
    def my_func(self):
        a = [1,2,3]
        b = np.random.normal(0,1,10000)
        result = []
        # we wish to parallelise over the array `a`
        for sub_array in np.array_split(a,3):
            result.append(self.func.remote(sub_array,b))
        return result

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐


依赖报错 idea导入项目后依赖报错,解决方案:https://blog.csdn.net/weixin_42420249/article/details/81191861 依赖版本报错:更换其他版本 无法下载依赖可参考:https://blog.csdn.net/weixin_42628809/a
错误1:代码生成器依赖和mybatis依赖冲突 启动项目时报错如下 2021-12-03 13:33:33.927 ERROR 7228 [ main] o.s.b.d.LoggingFailureAnalysisReporter : *************************** APPL
错误1:gradle项目控制台输出为乱码 # 解决方案:https://blog.csdn.net/weixin_43501566/article/details/112482302 # 在gradle-wrapper.properties 添加以下内容 org.gradle.jvmargs=-Df
错误还原:在查询的过程中,传入的workType为0时,该条件不起作用 <select id="xxx"> SELECT di.id, di.name, di.work_type, di.updated... <where> <if test=&qu
报错如下,gcc版本太低 ^ server.c:5346:31: 错误:‘struct redisServer’没有名为‘server_cpulist’的成员 redisSetCpuAffinity(server.server_cpulist); ^ server.c: 在函数‘hasActiveC
解决方案1 1、改项目中.idea/workspace.xml配置文件,增加dynamic.classpath参数 2、搜索PropertiesComponent,添加如下 <property name="dynamic.classpath" value="tru
删除根组件app.vue中的默认代码后报错:Module Error (from ./node_modules/eslint-loader/index.js): 解决方案:关闭ESlint代码检测,在项目根目录创建vue.config.js,在文件中添加 module.exports = { lin
查看spark默认的python版本 [root@master day27]# pyspark /home/software/spark-2.3.4-bin-hadoop2.7/conf/spark-env.sh: line 2: /usr/local/hadoop/bin/hadoop: No s
使用本地python环境可以成功执行 import pandas as pd import matplotlib.pyplot as plt # 设置字体 plt.rcParams['font.sans-serif'] = ['SimHei'] # 能正确显示负号 p
错误1:Request method ‘DELETE‘ not supported 错误还原:controller层有一个接口,访问该接口时报错:Request method ‘DELETE‘ not supported 错误原因:没有接收到前端传入的参数,修改为如下 参考 错误2:cannot r
错误1:启动docker镜像时报错:Error response from daemon: driver failed programming external connectivity on endpoint quirky_allen 解决方法:重启docker -> systemctl r
错误1:private field ‘xxx‘ is never assigned 按Altʾnter快捷键,选择第2项 参考:https://blog.csdn.net/shi_hong_fei_hei/article/details/88814070 错误2:启动时报错,不能找到主启动类 #
报错如下,通过源不能下载,最后警告pip需升级版本 Requirement already satisfied: pip in c:\users\ychen\appdata\local\programs\python\python310\lib\site-packages (22.0.4) Coll
错误1:maven打包报错 错误还原:使用maven打包项目时报错如下 [ERROR] Failed to execute goal org.apache.maven.plugins:maven-resources-plugin:3.2.0:resources (default-resources)
错误1:服务调用时报错 服务消费者模块assess通过openFeign调用服务提供者模块hires 如下为服务提供者模块hires的控制层接口 @RestController @RequestMapping("/hires") public class FeignControl
错误1:运行项目后报如下错误 解决方案 报错2:Failed to execute goal org.apache.maven.plugins:maven-compiler-plugin:3.8.1:compile (default-compile) on project sb 解决方案:在pom.
参考 错误原因 过滤器或拦截器在生效时,redisTemplate还没有注入 解决方案:在注入容器时就生效 @Component //项目运行时就注入Spring容器 public class RedisBean { @Resource private RedisTemplate<String
使用vite构建项目报错 C:\Users\ychen\work>npm init @vitejs/app @vitejs/create-app is deprecated, use npm init vite instead C:\Users\ychen\AppData\Local\npm-