RuntimeError: The following operation failed in the TorchScript interpreter. Traceback of TorchScript (most recent call last): RuntimeError: nvrtc: error: invalid value for --gpu-architecture (-arch)

这个错误通常发生在使用PyTorch和TorchScript进行GPU代码编译时。它表示你的GPU架构不被支持或不可用。

要解决此问题，你需要确保使用的GPU架构是支持的，并且已经正确安装了CUDA和cuDNN。你可以在NVIDIA的官方网站上了解你的GPU是否受支持，并获取正确的CUDA和cuDNN版本。此外，你还需要确保你的PyTorch版本与CUDA和cuDNN版本兼容。

如果你已经安装了正确的CUDA和cuDNN版本，并且你的GPU受支持，但仍然出现该错误，则可能是由于你的PyTorch版本不支持该GPU架构。在这种情况下，尝试使用较新的PyTorch版本或更改GPU架构以解决问题。

相关问题

terminate called after throwing an instance of 'std::runtime_error' what(): The following operation failed in the TorchScript interpreter. Traceback of TorchScript, serialized code (most recent call last): File "code/__torch__/models/yolo.py", line 33, in forward _22 = getattr(self.model, "2") _23 = getattr(self.model, "1") _24 = (getattr(self.model, "0")).forward(x, ) ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ <--- HERE _25 = (_22).forward((_23).forward(_24, ), ) _26 = (_20).forward((_21).forward(_25, ), ) File "code/__torch__/models/common.py", line 19, in forward _8 = torch.slice(_7, 3, 1, 9223372036854775807, 2) input = torch.cat([_2, _4, _6, _8], 1) return (_0).forward(input, ) ~~~~~~~~~~~ <--- HERE

这个错误提示来自C++的std::runtime_error，可能是由于TorchScript解释器出现了问题导致的。根据Traceback信息，可能是在yolo.py的第33行出现了问题，具体是在调用模型的某个属性时出现了异常。建议检查该属性是否正确定义并且被正确调用。同时，也可以尝试使用其他的调试方法，例如添加日志或者使用调试工具来定位问题。