Le-Xiaohuai-speech / DPCRN_DNS3

Implementation of paper "DPCRN: Dual-Path Convolution Recurrent Network for Single Channel Speech Enhancement"
188 stars 41 forks source link

How to set Multiple GPU's instead of cuda=0 or cuda=1 #24

Open ChandanaJonna opened 2 years ago

ChandanaJonna commented 2 years ago

I am training DPCRN with my data. I have 2 GPU's (GPU 0, GPU 1). But with args configuration I am able to choose either CUDA 0 or CUDA 1. With single GPU, my training speed to very slow. I would like to use both GPU's. How to set more than 1 GPU for parallel processing.

image

Kindly share your comments. Thanks in advance.