modify a python code to work with multi-GPU system

Hi,

I have a python script (very short - 10 lines or so) that do calculation in CPU and GPU.

I would like it to work with multi-GPU system.

for that you need a system to test this code. I will give you an AWS instace that yu will use.

deliverables:
1. change the code to fulfill calculating with multi-GPU system.
2. explain what you did.

Thanks