pytorch判断gpu是否可用

在 PyTorch 中，可以使用`torch.cuda.is_available()`函数来判断当前系统是否支持 GPU，并且 GPU 是否可用。下面是一个示例：

python
import torch

# 检查是否支持 GPU
if torch.is_available():
print("GPU 可用")
else:
print("GPU 不可用")

输出结果会根据系统的情况而定，如果 GPU 可用，则会打印"GPU 可用"，否则会打印"GPU 不可用"。

在使用 PyTorch 进行深度学习任务时，可以通过将模型和数据移动到 GPU 上来加速计算。可以使用`model.to('cuda')`将模型移动到 GPU 上，使用`tensor.to('cuda')`将张量移动到 GPU 上。这样，PyTorch 就会在 GPU 上执行相应的操作。