If you just call cuda , then the tensor is placed on GPU 0. The torch.nn.Module class also has to adnd cuda functions which puts the entire network on a ...
確定! 回上一頁