torch.cuda.stream(your_stream) 435 do what you want, I guess 
And you can use multiprocess 177 to launch the different process with different streams. Note that multiprocess with cuda is only suported in Python 3