PyTorch 1.0 中文文档：torch.cuda

译者：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 用来改变选择的设备。

阅读全文／改进本文

转载于:https://www.cnblogs.com/wizardforcel/p/10350594.html