地表最强Python并行加速Dask踩坑记录(一)

Dask的安装很容易直接pip install

https://docs.dask.org/en/latest/install.html

这上面介绍了三种安装方式,pip,conda和source

需要注意的是我是macOS, pip安装到python2.7 pip3安装到python3.6

我也用Conda装了我也有自己的conda,

但是我想Test却test不出来,因为我不知道Dask这个folder在哪里,我怎么cd到dask

可能如果要这样run下面的test只能从source来安装了

Test

Test Dask with py.test

//cd dask

//run py.test

算了。我们就假装是装好了的吧。开始学习!

Dask Tutorial 安装

首先我是来到了这个地址

https://tutorial.dask.org/00_overview.html

这里面缺了两个小步骤

正确的应该是5步骤一步不拉下

1. git clone http://github.com/dask/dask-tutorial

2. cd dask-tutorial

3.conda env create -f binder/environment.yml 

4.conda activate dark-tutorial

5. Jupyter notebook

就打开Jupyter Notebook了

下面这个步骤在Jupyter Notebook中进行

报错了我也不知道是不是对的。

好像是对的 delayed这个模块的作业做完了哈哈

我觉得这个tutorial非常interactive

还有思考题 上机作业题 很爽啊


下面进入作业题:

第一章:用dask.delayed并行加速

1.1 基础介绍

这里给读者简明扼要的介绍分布式是什么,用了一个例子的方式,很生动有趣。

举个例子,咱们用一个sleep函数,让函数执行的时候能休眠一秒钟

然后咱们计算结果z等于x加y,这里用到了上面的两个函数。同时还会用到Notebook里面一个简单的功能,也就是%%time,这个功能可以计算时间,看Wall time就可以了。

看看例子,怎么用Dask Delayed来并行化加速吧,只要加一个delayed再括号就可以。当然,在此之前需要“from dask import delayed”,最后输出z的时候加上“.compute()"就行

其实吧,这个z为什么要后面加个compute?

这是因为z是一个惰性对象(lazy object),当我们写了 “z=delayed(add)(x,y)”的时候,还没有计算,compute的时候才计算。这个概念如果了解Spark RDD的人应该比较清楚了哈哈。

比较有趣的是,dask还有一个z.visualize()函数可以输出计算图,有点像tensorflow的tensorboard。


灵魂拷问思考题:怎么加速啦?

第一个问题:上面例子中怎么做到从3s加速到2s的,我们能不能并行加速到1s呢?

第二个问题: 如果inc和add函数都没有sleep(1)这个东西会怎样?Dask还能不能加速这个代码呢?

第三个问题:如果我们有想访问x或者y的多个输出会怎么样?

第一题第二题我有答案,第三题其实不太会,欢迎讨论。。。

1.2 自己写一个for loop并行加速你会么?

没加速的程序都已经给你写好了,在下面。自己研究一下怎么用delayed做并行加速吧。照葫芦画瓢,其实就是想哪里加delayed的问题。是不是只要看到函数方法就加delayed呢,我们后面会回答这个问题。

我自己写的代码长这样,好像是对的。一开始我差点吧append 用了delayed。所以我觉得第二题里面如果没有sleep用了delayed好像时间不会变化,咱们在下一阶段作业的时候试试呗。



1.3 并行化加速控制流怎么做?

一般来讲我们不能照葫芦画瓢的把所有函数都delay。因为一般来说我们需要delay一些函数,而有些函数是要马上调用的。特别是快速call A函数能够决定我们是调用哪些slower的函数时。

Delay or not to delay, this is a problem!

生存与死亡是你的选择,delay还是不delay也是哈!需要仔细考虑。

在下面的例子中,如果输入是偶数,我们调用inc函数,如果是奇数,我们调用double。is_even这个函数在控制流语句中,需要立即调用(call immediately)如果加上delay会报错的!

错误玩法

下面这才是正确的玩法


灵魂拷问:控制流加速相关

1. 还有什么别的控制流我们不能用delayed么? 姐感觉while啥的也是哈

2.如果在is_even(x)这里用了delayed会怎么样? 前面演示了,会炸掉报错啊

3.为什么sum可以delay呢,sum也是计算函数而且运行的快啊? sum是比如A+B+C啥的,A B C不是并行计算的么,如果A和C算出来了不用等B可以先加的意思咯。


1.4 并行加速更复杂的代码

su

用dask.delayed来并行加速上述代码,还需要知道点别的储备知识。 

1.访问delayed对象的方法和属性会delay,所以如果你已经有了一个delayed object,就可以做数值计算,切片,方法调用等等。


2.单输出的时候 .compute() 方法还挺好用,但是多输出的时候就用dask.compute函数吧:


我是这么写的 写错了!

我还是不太懂两个点,一个是啥应该写delayed,啥不该写delayed

另外一个是怎么继承delayed这个属性啊

正确答案是这样

从pd read csv开始delayed 其他都是sequence可以一直继承下来的 ,就不用再次写delayed了

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。