Open delaossa opened 6 months ago
When passing use_cuda=True, gpu_id=1 to the Generator, still the gpu 0 is used.
use_cuda=True, gpu_id=1
Generator
@delaossa Could you post a script to reproduce the error? In addition, could you specify on which platform you were running this?
When passing
use_cuda=True, gpu_id=1
to theGenerator
, still the gpu 0 is used.