New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Jump to bottom

Device mismatch when using Adversarial Library attack #112

Open

tsalmonking opened this issue Feb 17, 2025 · 0 comments

tsalmonking commented Feb 17, 2025

When running examples on GPU instead of CPU, attacks from the adversarial library return the following error:

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda! (when checking argument target for method wrapped_CUDA_nll_loss_forward)

tsalmonking mentioned this issue

Correct device mismatch issue #113

Open

# for free to join this conversation on GitHub. Already have an account? # to comment