错误现象: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
9pip install --no-build-isolation -e .
--no-build-isolation:
这个选项告诉pip在构建包时不要使用隔离环境。默认情况下,pip在构建包时会创建一个隔离环境,这意味着构建过程中不会访问系统的全局Python环境。
使用这个选项可以让构建过程访问当前环境中的其他包,这对于依赖于当前环境中的特定库或版本的情况非常有用。
-e:
这个选项表示“editable”模式,即以可编辑的方式安装包。这意味着安装的包可以在不重新安装的情况下直接修改源代码,并且这些修改会立即生效。
这对于开发阶段特别有用,因为你可以随时修改代码并测试更改的效果,而不需要每次都重新安装包。