[cifar ds training]: Set cuda device during initialization of distributed backend.#931