它会跟踪当前选定的GPU,并且默认情况下会在该设备上创建您分配的所有CUDA tensors。可以使用 torch.cuda.device 上下文管理器更改所选设备。 但是,一旦分配了tensor,就 ...
確定! 回上一頁