转载请注明作者梦里茶
需求
在研究中遇到这样一种需求，
- 在Matlab端做GPU运算；
- 在Python端做数据清洗和数据分析；
- 两端分属两个服务器；
- M端需要等待P端完成数据清洗才能开始做训练和预测
- P端需要等待M端完成训练和预测才能做数据分析
问题
- 数据很多，不适合将两端合并到一个服务器上
- 存在增量训练，对同一份数据，可能要结合多份额外数据进行N次传输
- 由于采用10折交叉验证，数据划分为10份，总共要进行10×N次传输
- M端和P端都需要各自运行一段时间才能出结果，需要等待，说快不快，说慢不慢
上面的几个点使得训练的过程非常繁琐耗时
探索
作为不耐心而且懒的程序员，超过15秒的事情就想要自动化，于是开始想能不能把上面这个过程自动化
方案一 用一种语言重写另一方的代码
- Python端逻辑复杂，与数据关联密切，重构为Matlab代码工程量大（而且我还是比较熟悉Python）
- Matlab端是某篇论文的代码，使用了MatConvNet这个库做CNN，用Python实现工程量也很大，而且因为是别人的代码，还可能有很多未知的坑
方案二 使用第三方库作为Lib相互调用
- 由于Matlab端涉及GPU运算，配置比较复杂
- 作为库相互调用使得两个模块的耦合增加，不符合解耦原则
- 需要编写接口代码
方案三 Matlab和Python进行进程间通信
- 进程间通信要求两边各有一个可执行程序，但是Matlab打包麻烦（需要下另一个收费的打包工具）具体看这里，在GUI界面中操作相对方便，并且打包后不好调试
- 远程进程间通信有许多方案可以选择，例如socket，这个工作量比上面的方案小，但也需要额外编写一部分功能代码
方案四 (Final)使用Matlab进行执行shell命令进行scp，以文件的方式进行数据传输
- 改动的代码最少
- 依旧是两个模块，依旧低耦合
- 通信性能没有库调用或者Socket那么好，但程序主要性能瓶颈不在数据传输，而在于训练和预测，所以没关系
- 执行ssh的时候可以同时执行另一个服务器上的命令，从而调用Python端脚本进行Python端工作
下面具体讲方案四的实现
实现
Matlab执行shell命令
setenv('d_name', dataset_name);
setenv('std_idx', num2str(i));
% transport raw image predict data
!env LD_LIBRARY_PATH='' scp ${d_name}_train.log cwh@192.168.231.171:/home/cwh/coding/Project/data/${std_idx}-train/renew_pid.log
!env LD_LIBRARY_PATH='' ssh cwh@192.168.231.171 "cd /home/cwh/coding/Project; python data_analysis.py data/${std_idx}-train"
其中
- 通过setenv来设置环境变量，从而动态决定要执行的脚本的内容
- 在matlab代码中，使用
!开头，代表这是一行shell命令 - 需要把LD_LIBRARY_PATH这个环境变量置为空，防止使用matlab自带的一些lib，导致ssh等命令执行失败
SSH
- Matlab是有一个库可以做ssh和scp的
- 但是，不想把密码写在代码里，并且懒得去用第三方的东西
- 于是想直接用shell调用
- shell调用有一个老大难的问题，就是需要输密码
- 受Hadoop分布式配置的启发，发现可以做ssh免密码配置，避开输密码的问题
- 免密码具体参考这个教程
- ssh配置免密之后，scp也是免密的，另外还有一个好处，ssh可以通过
""来嵌入ssh之后要执行的代码，从而做Python调用
Python
- 如果之前Python端是用IDE开发的，转为shell调用就要注意两个问题
- 路径，cd到正确的目录，Python端的文件操作和lib import的相对路径是相对于要执行的py文件
- 库
# 加载对应的库 env LD_LIBRARY_PATH='/your/lib/path/sth.so'- 命令行参数，参考这个教程
同步控制
- 由于matlab调用shell命令时会陷入等待，会等待命令完成再执行下一步，不像socket发完消息后不知道什么时候另一边运行结束，所以不需要自己实现排队等待同步互斥的东西
- 如果需要在等待的时候做些并发，可以考虑matlab 并行工具箱，结合Python端的并发几乎就是一个分布式框架了，有空再深入去搞一搞
SUMMARY
- matlab调shell
- shell中ssh做远程进程调用
- scp做数据传输
- 效果：尽可能轻量地解决自动化的问题
网友评论