复现论文时设置CUDA_HOME仍无法安装CUDA扩展，请求解决

问题场景 #

执行pip install .安装本地CUDA扩展包时，报错OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.，已添加环境变量但问题依旧。

解决步骤 #

• 确认CUDA_HOME路径正确性
  找到你的CUDA安装根目录（默认路径如C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8，vX.X为具体版本号），确保环境变量设置的是这个绝对路径，路径中即使包含空格也不要添加引号。

• 验证环境变量是否生效
  关闭当前PowerShell窗口，重新打开后执行：

  echo $env:CUDA_HOME
  

  若输出正确的CUDA安装路径，说明环境变量生效；否则需重新配置环境变量并重启终端。

• 检查PyTorch与CUDA版本兼容性
  执行以下命令查看PyTorch绑定的CUDA版本：

  python -c "import torch; print(torch.version.cuda)"
  

  确保你安装的CUDA Toolkit版本与PyTorch的CUDA版本完全一致，版本不匹配会导致PyTorch无法识别CUDA环境。

• 手动指定CUDA路径执行安装
  若环境变量仍未被识别，可在安装前临时指定CUDA_HOME：

  $env:CUDA_HOME="你的CUDA安装绝对路径"
  pip install .
  

  替换路径为你实际的CUDA根目录（如C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8）。

• 验证CUDA是否正确安装
  执行以下命令检查CUDA是否正常安装：

  nvcc --version
  

  若无法输出版本信息，说明CUDA Toolkit未正确安装，需重新下载对应版本的CUDA Toolkit并完成安装。

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