PyTorch官方中文文档:torch.cuda

torch.cuda 该包增长了对CUDA张量类型的支持,实现了与CPU张量相同的功能,但使用GPU进行计算。html 它是懒惰的初始化,因此你能够随时导入它,并使用is_available()来肯定系统是否支持CUDA。python CUDA语义中有关于使用CUDA的更多细节。函数 torch.cuda.current_blas_handle() 返回cublasHandle_t指针,指向当前c
相关文章
相关标签/搜索