怎么默认使用核显,不使用集显
时间: 2024-02-08 21:03:12 浏览: 290
itext导出pdf不显示中文 ITextRenderer不显示中文
如果你想让程序默认使用核显而不是集显,你需要先确定你的显卡是不是带有独立显卡和集成显卡的双显卡系统。如果是,你需要指定使用独立显卡的设备 id,通常是一个较大的整数,比如 `1` 或 `2`。
在 PyTorch 中,可以使用 `torch.cuda.device_count()` 函数获取可用的 GPU 数量,然后使用 `torch.cuda.get_device_properties()` 函数获取每个 GPU 的属性信息,包括设备名称、总内存、设备 id 等。你可以根据这些信息,选择合适的设备。
以下是一段代码示例,用于选择第一个可用的 GPU 设备,但排除掉设备名称中包含 "Intel" 的设备,即排除集成显卡:
```python
import torch
def get_device():
if torch.cuda.is_available():
num_gpus = torch.cuda.device_count()
for i in range(num_gpus):
gpu_props = torch.cuda.get_device_properties(i)
if 'Intel' not in gpu_props.name:
return torch.device(f'cuda:{i}')
return torch.device('cpu')
device = get_device()
print(device)
```
这个函数会返回第一个找到的非集成显卡设备,如果没有找到,则返回 CPU 设备。你可以将这个函数集成到你的代码中,以获取合适的设备。
阅读全文