RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Float'
时间: 2023-12-04 13:04:56 浏览: 70
Python RuntimeError: thread.__init__() not called解决方法
这个错误通常是由于 PyTorch 的版本与 CUDA 版本不兼容导致的。尝试降低 PyTorch 的版本,或者更新 CUDA 驱动程序以解决此问题。
另外,您也可以尝试在训练模型时使用 CPU 而不是 GPU,看看是否会有同样的错误。如果不会出现错误,那么问题可能是与 GPU 相关的。您可以通过检查 CUDA 版本、GPU 驱动程序和 PyTorch 版本来进一步调查问题。
阅读全文