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