DEV Community
·
解决ContourCraft中的CUDA版本和GPU架构问题
💡
原文英文，约500词，阅读约需2分钟。
📝
内容提要
在Windows 11、RTX 4090和WSL2上安装CCCollisions模块时，遇到CUDA版本不匹配和GPU架构不支持的问题。通过安装CUDA 11.7并设置环境变量，成功解决了这些问题，顺利完成安装。
🎯
关键要点
- 在Windows 11、RTX 4090和WSL2上安装CCCollisions模块时遇到CUDA版本不匹配和GPU架构不支持的问题。
- 错误1：CUDA版本不匹配，环境中安装的CUDA版本为12.6，而PyTorch编译时使用的是11.7。
- 解决方案：重新初始化WSL2环境并安装CUDA 11.7，确保CUDA版本匹配。
- 错误2：不支持的GPU架构，CUDA 11.7不识别RTX 4090的compute_89架构。
- 解决方案：将架构设置为CUDA 11.7支持的compute_86，并设置相应的环境变量。
- 临时修复：通过命令设置环境变量以支持正确的CUDA架构。
- 永久修复：将环境变量添加到shell配置文件中，并重新登录以应用更改。
- 成功安装CCCollisions模块，未再出现问题。
❓
延伸问答
在安装CCCollisions模块时遇到的CUDA版本不匹配问题是什么？
CUDA版本不匹配是指环境中安装的CUDA版本为12.6，而PyTorch编译时使用的是11.7，导致运行时错误。
如何解决CUDA版本不匹配的问题？
可以通过重新初始化WSL2环境并安装CUDA 11.7来解决CUDA版本不匹配的问题。
RTX 4090的GPU架构不被CUDA 11.7支持的原因是什么？
因为RTX 4090支持compute_89架构，而CUDA 11.7不识别该架构，导致构建过程中的致命错误。
如何设置CUDA架构以支持RTX 4090？
可以将架构设置为CUDA 11.7支持的compute_86，并通过命令设置相应的环境变量。
临时修复CUDA架构不支持的方法是什么？
临时修复方法是通过命令设置环境变量，例如export TORCH_CUDA_ARCH_LIST="8.6"。
如何永久修复CUDA架构不支持的问题？
可以将环境变量添加到shell配置文件中，并重新登录以应用更改。
🏷️
标签
➡️
