【报错系列】autodl运行python setup.py install安装tinycudann包时遇到OSError: Unknown compute capability.报错

【报错系列】autodl运行python setup.py install安装tinycudann包时遇到OSError: Unknown compute capability.报错

【先赞后看养成习惯】求关注+点赞+收藏
问题描述：autodl运行python setup.py install安装tinycudann包时，遇到报错：OSError: Unknown compute capability. Specify the target compute capabilities in the TCNN_CUDA_ARCHITECTURES environment variable or install PyTorch with the CUDA backend to detect it automatically.

问题分析：这个错误是因为在编译PyTorch扩展时，系统无法自动检测到你的GPU的计算能力。解决这个问题的方法是手动指定GPU的计算能力。

解决方法：
首先，需要知道你的GPU的计算能力。您可以在NVIDIA的官方网站上查询到这个信息。然后，将这个值乘以10。
例如，如果您的GPU是4090，其计算能力为8.6，那么您需要在命令行中输入以下命令

export TCNN_CUDA_ARCHITECTURES=86

这样，当你再次编译PyTorch扩展时，系统就能正确识别您的GPU的计算能力了。
另外，如果你的CUDA版本和GPU的计算能力不匹配，可能也会出现类似的问题。在这种情况下，你需要安装与您的GPU计算能力匹配的CUDA版本