raise AssertionError("Torch not compiled with CUDA enabled") AssertionError: Torch not compiled with CUDA enabled
时间: 2023-11-08 09:03:54 浏览: 204
scrcpy投屏 AssertionError: java.lang.reflect.InvocationTargetExcep
5星 · 资源好评率100%
这个错误提示表明你在使用Torch时启用了CUDA,但是你的Torch没有编译支持CUDA。要解决这个问题,你可以尝试以下几个步骤:
1. 确保你的GPU驱动程序已正确安装并与CUDA版本兼容。
2. 确保你已正确安装了CUDA工具包,并且版本与你的PyTorch版本兼容。
3. 检查你的PyTorch安装是否支持CUDA。你可以通过在Python中运行以下代码来检查:
```
import torch
print(torch.cuda.is_available())
```
如果返回False,则表示你的PyTorch没有支持CUDA编译。
如果上述步骤都正确无误,并且问题仍然存在,那可能是因为你的PyTorch版本不支持CUDA编译。尝试安装与你的PyTorch版本相对应的CUDA版本,或者尝试降低PyTorch版本以适应你的CUDA版本。
阅读全文