如何判断是否有CUDA?

共计 511 个字符,预计需要花费 2 分钟才能阅读完成。

如何判断是否有 CUDA?

在 Python 中,我们可以使用 torch.cuda.is_available() 函数来检查 CUDA 是否可用。该函数返回一个布尔值,表示 CUDA 是否可用。

使用 torch.cuda.is_available() 函数

要判断是否有 CUDA,可以按照以下步骤:

  1. 导入 torch 库:import torch
  2. 调用 torch.cuda.is_available() 函数:cuda_available = torch.cuda.is_available()
  3. 根据返回值判断 CUDA 是否可用:
    • 如果 cuda_available 为 True,则表示 CUDA 可用。
    • 如果 cuda_available 为 False,则表示 CUDA 不可用。

下面是一个示例代码:

import torch

cuda_available = torch.cuda.is_available()

if cuda_available:
print(“CUDA 可用 ”)
else:
print(“CUDA 不可用 ”)

运行以上代码,将会根据 CUDA 的可用性输出相应的结果。

结论

通过调用 torch.cuda.is_available() 函数,我们可以判断 CUDA 是否可用,从而在需要使用 CUDA 加速的时候进行相应的处理。


正文完