search

ucbrise / piranha

Piranha: A GPU Platform for Secure Computation
MIT License
92 stars 26 forks source link

Run piranha on a single GPU machine with multiple GPUs #1

Closed DylanWangWQF closed 2 years ago

DylanWangWQF commented 2 years ago

Hi Jean-Luc @jlwatson , As introduced, piranha runs on 2/3/4 machines in the AWS cluster. However, currently, I have only one GPU machine with 4 Tesla GPUs or some GPU machines with different hardware configurations. Does piranha provide a simulation for running it on a single GPU machine with multiple GPUs？ Thanks in advance!

jlwatson commented 2 years ago

Hey, great question! I do have a slightly janky setup for running the code on a single machine with multiple GPUs that I'll clean up and push in the next day or so. It'll let you set which GPU to run each party on -- I'll include some instructions too :)

DylanWangWQF commented 2 years ago

Looking forward to this nice update! I believe this is what some people expect.

jlwatson commented 2 years ago

#8fd8219 should address this; README updated to give some clear instructions for running on localhost. You can modify which GPUs without code changes by limiting which GPU each process sees using CUDA_VISIBLE_DEVICES.

Feel free to PR if you run into any issues and the instructions need an update :)