Windows环境下PyTorch无法检测CUDA的GPU配置咨询

我拥有NVIDIA GeForce RTX 3070显卡，当前环境版本如下：

windows 10 Enterprise
python 3.8.10
torch 1.11.0

执行nvcc -V命令得到：

Cuda Compliation tools, release 11.3, V11.3.58

执行nvidia-smi命令得到：

NVIDIA-SMI 465.89 Driver Version: 465.89 CUDA Version 11.3

运行以下代码：

import torch
print(torch.cuda.is_available())

返回结果为False，请问在Windows系统下需要进行哪些配置才能使用GPU？

• 匹配PyTorch与CUDA版本
  你当前安装的PyTorch 1.11.0需对应CUDA 11.3版本的支持包，而非CPU版本。打开命令行执行以下命令重新安装：

  pip3 install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113
  

• 配置系统环境变量
  确保CUDA路径已添加至系统环境变量：

  • 右键「此电脑」→「属性」→「高级系统设置」→「环境变量」
  • 系统变量中确认CUDA_PATH值为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3（默认安装路径）
  • 将C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin和C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\libnvvp添加到Path系统变量
  • 重启命令行或电脑后，执行echo %CUDA_PATH%验证路径配置生效

• 确认驱动兼容性
  当前465.89版本驱动兼容CUDA 11.3，若驱动存在损坏可重新安装同版本或NVIDIA官方推荐的CUDA 11.3适配驱动，无需盲目升级更高版本。

• 验证PyTorch CUDA支持
  重新安装后运行以下代码确认配置成功：

  import torch
  print(torch.version.cuda)  # 应输出11.3
  print(torch.cuda.is_available())  # 应返回True
  print(torch.cuda.get_device_name(0))  # 应显示NVIDIA GeForce RTX 3070
  

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