python - MPI を使用した Python での画像操作

Question

Python で MPI (mpi4py) を使用して画像操作をしようとしています。これまでの私のコードは次のとおりです。

from mpi4py import MPI
from PIL import Image

comm = MPI.COMM_WORLD
size = comm.size #number of processors
rank = comm.rank #calling process rank
root = 0
if rank == 0:
    print "rank = 0 \n"
    img = Image.open('dog.jpg')
    width, height = img.size
    pix = list(img.getdata())
    for i in range(len(pix)):
        pix[i] = list(pix[i])
    chunks = [[] for _ in range(size)]
    for i, chunk in enumerate(pix):
        chunks[i % size].append(chunk)
else:
    print "rank != 0"

data = comm.scatter(chunks, root)
#do some pixel manipulations
data = comm.gather(data, root)
for i in range(len(pix)):
    pix[i] = tuple(data[0][i])
image = Image.new("RGB", (width, height))
image.putdata(pix)
image.save("test.jpg")

で実行しようとすると

mpirun -np 1 python parallel.py

それは正常に動作します。しかし、プロセッサの数を増やそうとすると、次のエラーが発生します

mpirun -np 3 python parallel.py
rank = 1
rank = 2
rank = 0 

Traceback (most recent call last):
  File "parallel.py", line 21, in <module>
    data = comm.scatter(chunks, root)
NameError: name 'chunks' is not defined
Traceback (most recent call last):
  File "parallel.py", line 21, in <module>
    data = comm.scatter(chunks, root)
NameError: name 'chunks' is not defined

Virtual Box を使用して、Ubuntu 仮想マシンでコードを実行しています。あなたの助けに感謝！