Python中无法检测NVIDIA RTX 4080笔记本GPU用于CUDA运算的求助

核心问题 #

你当前安装的PyTorch是仅CPU版本（2.2.0+cpu），这是无法调用CUDA GPU的根本原因，必须替换为适配CUDA的版本。

解决步骤 #

1. 卸载现有CPU版PyTorch #

打开命令提示符（CMD）或终端，执行对应命令：

• 使用pip卸载：

  pip uninstall torch torchvision torchaudio -y
  

• 使用conda卸载（如果用Anaconda/Miniconda环境）：

  conda uninstall torch torchvision torchaudio -y
  

2. 安装适配CUDA的PyTorch版本 #

首先确认你的NVIDIA驱动支持的CUDA版本：RTX 4080笔记本GPU推荐使用CUDA 11.8或12.1，先通过NVIDIA GeForce Experience把驱动更新到最新版本。

然后根据选择的CUDA版本，执行对应的安装命令：

• 安装支持CUDA 11.8的PyTorch 2.2.0：

  pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu118
  

• 安装支持CUDA 12.1的PyTorch 2.2.0：

  pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu121
  

3. 验证安装结果 #

运行以下Python代码，确认GPU被识别：

import torch
print("CUDA是否可用:", torch.cuda.is_available())
print("GPU数量:", torch.cuda.device_count())
if torch.cuda.is_available():
    print("GPU名称:", torch.cuda.get_device_name(0))

正常情况下会输出CUDA是否可用: True，并显示你的RTX 4080笔记本GPU名称。

4. 额外检查项 #

• 确认笔记本已设置为“独立显卡”模式并重启生效
• 确保你使用的Python环境是安装了CUDA版PyTorch的环境（可通过python --version和pip list | findstr torch验证）
• 打开命令提示符输入nvidia-smi，检查是否能正常显示GPU信息及支持的CUDA版本

内容的提问来源于stack exchange，提问作者AllramEst