GPU并行
类似Joblib的接口,用于并行GPU计算(例如,数据预处理)。
安装
python3 -m pip install gpuparallel
# or
python3 -m pip install git+git://github.com/vlivashkin/gpuparallel.git
基本用法
from gpuparallel import GPUParallel
from functools import partial
def perform ( idx , worker