search

Trusted-AI / adversarial-robustness-toolbox

Adversarial Robustness Toolbox (ART) - Python Library for Machine Learning Security - Evasion, Poisoning, Extraction, Inference - Red and Blue Teams
https://adversarial-robustness-toolbox.readthedocs.io/en/latest/
MIT License
4.75k stars 1.15k forks source link

GPU support for MembershipInferenceBlackBox#infer #899

Closed evansakmar closed 3 years ago

evansakmar commented 3 years ago

Describe the bug Whenever you run infer on MembershipInferenceBlackBox you get a CUDA runtime error.

The fit method seems to handle CUDA properly already. Noting the _tocuda method.

To Reproduce Steps to reproduce the behavior:

Expected behavior It should run properly without erroring out

Stacktrace RuntimeError: cuda runtime error (77) : an illegal memory access was encountered at /pytorch/aten/src/THC/THCCachingHostAllocator.cpp:278

System information (please complete the following information):

evansakmar commented 3 years ago

I also have an incoming PR for this

beat-buesser commented 3 years ago

Hi @ehsankf Thank you very much!

ehsankf commented 3 years ago

@beat-buesser

Hello,

This issue has not been opened by me.

beat-buesser commented 3 years ago

Hi @ehsankf Thank you for notifying me, it must have been an autocomplete accident with user names starting with letter e. I wanted to mention @evansakmar above.