译者: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 ,那么它不会起任何作用。 |
---|
原文地址:https://www.cnblogs.com/wizardforcel/p/10350594.html
时间: 2024-11-08 22:39:30