PyTorch 1.0 中文文档：torch.cuda

Posted 2021-02-07 wizardforcel

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了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 用来改变选择的设备。

参数:	device (torch.device 或者 int) – 要选择的设备索引。如果这个参数是负数或者是 `None`，那么它不会起任何作用。

阅读全文／改进本文

以上是关于PyTorch 1.0 中文文档：torch.cuda的主要内容，如果未能解决你的问题，请参考以下文章