译者：bdqfork
这个包添加了对CUDA张量类型的支持，它实现了与CPU张量同样的功能，但是它使用GPU进计算。
它是懒加载的，所以你可以随时导入它，并使用 is_available() 来决定是否让你的系统支持CUDA。
CUDA semantics 有关于使用CUDA更详细的信息。
torch.cuda.current_blas_handle()返回一个cublasHandle_t指针给当前的cuBLAS处理。
torch.cuda.current_device()返回当前选择地设备索引。
torch.cuda.current_stream()返回当前选择地 Stream。
class torch.cuda.device(device)Context-manager 用来改变选择的设备。
| 参数: |
device (torch.device 或者 int) – 要选择的设备索引。如果这个参数是负数或者是 None，那么它不会起任何作用。 |
|---|