torch.cuda.is_available() 即使在使用 CUDA 安装 PyTorch 后也会返回 False

Question

我最近在我的机器上安装了支持 CUDA 的 PyTorch，但是当我运行

torch.cuda.is_available()

时，它返回

False

。我使用

nvidia-smi

验证了我的 GPU 设置，看来我的系统可以正确识别 GPU。

以下是我到目前为止所遵循的步骤：

1. 使用命令

   pip install torch torchvision torchaudio

   安装了支持 CUDA 的 PyTorch。

2. 确认我的系统已使用

   nvidia-smi

   识别我的 GPU。

3. 已验证是否安装了正确的 CUDA 版本。

尽管如此，运行

torch.cuda.is_available()

仍然会返回

False

。导致此问题的原因是什么？我该如何解决？

附加信息：

• 操作系统：Windows 11

• Python版本：3.10

• PyTorch 版本：2.3.1

• CUDA版本：12.5.82

• GPU型号：NVIDIA GeForce RTX 4060

我尝试过的：

• 重新安装具有 CUDA 支持的 PyTorch。

• 更新 GPU 驱动程序。

• 重新启动系统。

import torch

print(torch.cuda.is_available())  # Returns False
print(torch.cuda.current_device())  # Raises RuntimeError: No CUDA GPUs are available

我最近在我的机器上安装了支持 CUDA 的 PyTorch，但是当我运行

torch.cuda.is_available()

时，它返回

False

。

Answer 1

转到 Windows 图形设置，浏览要使用 CUDA 运行的 .py 文件，并将其设置为使用高性能 Nvidia 显卡。