pytorch测试GPU是否可用

检测GPU是否可用

要检查GPU是否可用，可以使用torch.cuda.is_available()函数。这个函数会返回一个布尔值，如果系统中有可用的GPU，它将返回True，否则返回False。以下是如何使用这个函数的示例代码：

import torch
print("torch版本:",torch.__version__)
print("CUDA是否可用:", torch.cuda.is_available()) 
print("CUDA版本:", torch.version.cuda)
print("cuDNN 是否启用:", torch.backends.cudnn.enabled)