torch.cuda会记录当前选择的GPU，并且分配的所有CUDA张量将在上面创建。可以使用torch.cuda.device上下文管理器更改所选设备。
但是，一旦张量被分配，您可以直接对其进行操作，而不考虑所选择的设备，结果将始终放在与张...
AI君
1年前 (2019-03-28) 104℃
0喜欢
PyTorch是使用GPU和CPU优化的深度学习张量库。
说明
自动求导机制
CUDA语义
扩展PyTorch
多进程最佳实践
序列化语义
Package参考
torch
torch.Tensor
torch.Storage
torc...
AI君
1年前 (2019-03-28) 246℃
0喜欢
