该包增加了对CUDA张量类型的支持，实现了与CPU张量相同的功能，但使用GPU进行计算。
它是懒惰的初始化，所以你可以随时导入它，并使用is_available()来确定系统是否支持CUDA。
CUDA语义中有关于使用CUDA的更多细节。
...
AI君
1年前 (2019-03-28) 125℃
0喜欢
torch.cuda会记录当前选择的GPU，并且分配的所有CUDA张量将在上面创建。可以使用torch.cuda.device上下文管理器更改所选设备。
但是，一旦张量被分配，您可以直接对其进行操作，而不考虑所选择的设备，结果将始终放在与张...
AI君
1年前 (2019-03-28) 96℃
0喜欢
