pycharm Torch not compiled with CUDA enabled
时间: 2024-05-28 09:07:50 浏览: 190
快速安装Pytorch以及CUDA匹配问题
5星 · 资源好评率100%
这个错误通常是因为你的PyTorch没有编译支持CUDA。 CUDA是NVIDIA提供的一种并行计算平台和编程模型,它可以使用GPU进行高性能计算。 在PyTorch中,如果你想使用GPU加速模型训练,你需要使用支持CUDA的PyTorch版本,并且你需要在安装PyTorch时选择相应的CUDA版本。
如果你已经安装了支持CUDA的PyTorch版本,但仍然遇到这个问题,可能是因为你的CUDA驱动程序版本与PyTorch不兼容。 你可以尝试升级或降级你的CUDA驱动程序版本,以使其与PyTorch兼容。
如果你不需要使用GPU加速训练,可以将代码中所有的`torch.cuda()`替换为`torch.cpu()`来强制在CPU上运行代码。
阅读全文