关于使用多GPU运行ann_to_snn.py

[ryikl]

你好，由于显存不够，我考虑使用多GPU运行ann_to_snn.py，一开始在device中写上要用的卡，但后边发现只用了一块卡。后来通过使用nn.DataParallel之后，会出现内存分布问题。我在一个issues中看到你聊到使用多GPU的问题，想请教一下你是怎么使用多GPU的，具体在哪里做改动。