[已解决]判断有否gpu，如有gpu让whisper用gpu，否则就用cpu

blackantt

判断有否gpu，如有gpu让whisper用gpu，否则就用cpu

最佳答案

月排行榜 / 总排行榜

FishC

2024-5-13 11:47:31

要判断计算机是否有GPU并根据情况选择使用GPU或CPU，可以使用Python中的`torch.cuda.is_available()`函数来检查是否存在可用的GPU。

1. import torch
   
2. 
   
3. if torch.cuda.is_available():
   
4.     device = torch.device("cuda")  # 如果存在GPU则选择使用GPU
   
5. else:
   
6.     device = torch.device("cpu")   # 否则选择使用CPU
   
7. 
   
8. print("当前设备:", device)
   
9. 
   
10. # 在后续代码中使用device来指定计算设备
    
11. # 例如：
    
12. # tensor = tensor.to(device)
    
13. # model = model.to(device)
    
14. # ...

复制代码

这样，如果计算机有GPU，程序将会在GPU上运行，否则会在CPU上运行。你可以根据具体需求，在需要使用GPU的地方将数据和模型转移到GPU上进行计算，而不需要担心计算机是否支持GPU的问题。

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。

跳转到最佳答案楼层

FishC

要判断计算机是否有GPU并根据情况选择使用GPU或CPU，可以使用Python中的`torch.cuda.is_available()`函数来检查是否存在可用的GPU。

1. import torch
   
2. 
   
3. if torch.cuda.is_available():
   
4.     device = torch.device("cuda")  # 如果存在GPU则选择使用GPU
   
5. else:
   
6.     device = torch.device("cpu")   # 否则选择使用CPU
   
7. 
   
8. print("当前设备:", device)
   
9. 
   
10. # 在后续代码中使用device来指定计算设备
    
11. # 例如：
    
12. # tensor = tensor.to(device)
    
13. # model = model.to(device)
    
14. # ...

复制代码

这样，如果计算机有GPU，程序将会在GPU上运行，否则会在CPU上运行。你可以根据具体需求，在需要使用GPU的地方将数据和模型转移到GPU上进行计算，而不需要担心计算机是否支持GPU的问题。

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。

civilfem

方法适用