pytorch查看gpu是否可用
时间: 2024-10-24 18:12:14 浏览: 15
Pytorch 实现GPU加速.zip
在PyTorch中,你可以通过`torch.cuda.is_available()`函数来检查当前计算机是否配备了GPU,并且该设备是否可以用于计算操作。这个函数会返回一个布尔值,如果可用则返回True,否则返回False。
如果你想要获取更详细的信息,比如GPU的品牌、型号、显存大小等,可以使用`torch.cuda.get_device_name()`获取设备名称,`torch.cuda.device_count()`获取可用的GPU数量,以及`torch.cuda.memory_allocated()`或`torch.cuda.memory_cached()`获取特定GPU的内存信息。
下面是一个简单的示例:
```python
import torch
# 检查是否有可用的GPU
if torch.cuda.is_available():
print("有GPU可用")
device = torch.device('cuda') # 使用GPU
else:
print("没有GPU,将使用CPU")
device = torch.device('cpu')
# 获取设备详细信息
device_name = torch.cuda.get_device_name(device.index)
print(f"使用的GPU是: {device_name}")
```
阅读全文