Skip to content

Can multi-GPU training be used to attack the dataset?/是否可以使用多GPU对数据集进行攻击 #163

Description

@cgdrm

Hello author, thank you for your wonderful work and contribution. When I attack graph data, I often encounter the problem of insufficient GPU memory, such as the Pubmed dataset. I would like to ask if it is possible to use multiple GPUs for the attack, and if so, how should I operate?
您好作者,感谢您的精彩工作和贡献。我在攻击图数据的时候经常遇到GPU内存不足的问题,例如Pubmed数据集。我想问是否可以使用多个GPU进行攻击,可以的话应该怎么操作?

RuntimeError: CUDA out of memory. Tried to allocate 24.00 MiB (GPU 0; 15.89 GiB total capacity; 14.17 GiB already allocated; 23.12 MiB free; 15.21 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions