Windows11下大模型环境报错 OSError: CUDA_HOME environment variable is not set | AI快速入门 2024

错误现象:OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.
解决办法:

  • 下载CUDA12.4
  • 解压安装,C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4
  • 配置环境变量,CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4
  • 进入python虚拟环境执行
    1
    2
    3
    4
    5
    6
    7
    8
    9
    pip install --no-build-isolation -e .

    --no-build-isolation:
    这个选项告诉pip在构建包时不要使用隔离环境。默认情况下,pip在构建包时会创建一个隔离环境,这意味着构建过程中不会访问系统的全局Python环境。
    使用这个选项可以让构建过程访问当前环境中的其他包,这对于依赖于当前环境中的特定库或版本的情况非常有用。

    -e:
    这个选项表示“editable”模式,即以可编辑的方式安装包。这意味着安装的包可以在不重新安装的情况下直接修改源代码,并且这些修改会立即生效。
    这对于开发阶段特别有用,因为你可以随时修改代码并测试更改的效果,而不需要每次都重新安装包。

参考文章:
OSError: CUDA_HOME environment variable is not set.