PyTorch官方中文文檔：torch.cuda

torch.cuda 該包增長了對CUDA張量類型的支持，實現了與CPU張量相同的功能，但使用GPU進行計算。html 它是懶惰的初始化，因此你能夠隨時導入它，並使用is_available()來肯定系統是否支持CUDA。python CUDA語義中有關於使用CUDA的更多細節。函數 torch.cuda.current_blas_handle() 返回cublasHandle_t指針，指向當前c

>>阅读原文<<