RuntimeError: ./mmcv/ops/csrc/pytorch/cuda/sparse_indice.cu 126
时间: 2023-09-30 07:11:59 浏览: 421
pytorch模型提示超出内存RuntimeError: CUDA out of memory.
5星 · 资源好评率100%
这个错误通常是由于CUDA版本不匹配导致的。您需要确保您使用的CUDA版本与您的mmcv库兼容。您可以尝试升级或降级CUDA版本,或者安装与您的CUDA版本匹配的mmcv版本。您还可以尝试更新您的NVIDIA驱动程序,以确保其与您的CUDA版本兼容。如果这些解决方案都不起作用,您可以尝试在GitHub上搜索相关问题或联系mmcv库的开发人员进行支持。
阅读全文