he-transformer with the GPU

intel / he-transformer

nGraph-HE: Deep learning with Homomorphic Encryption (HE) through Intel nGraph

Apache License 2.0

172 stars 35 forks source link

Open acoj1993 opened 3 years ago

acoj1993 commented 3 years ago

Hey @fboemer,

Is it possible to run the library with the GPU? If yes, what changes do I have to make?

Cheers, Aleksandar

fboemer commented 3 years ago

Hi Aleksandar, thanks for your interest. Unfortunately this library supports CPU only.

acoj1993 commented 3 years ago

Aha, ok, understood.

acoj1993 commented 3 years ago

While I am at it, another question.

Have you tried other ImageNet models, apart from MobileNet V2? Do you have any instructions or something I should keep in mind?