It keeps track of the currently selected GPU, and all CUDA tensors you ... device=cuda) # transfers a tensor from CPU to GPU 1 b = torch.tensor([1., 2.]) ...
確定! 回上一頁