电脑中没有cuda,只有cpu,并且安装了pytorch-cpu为什么还出现 raise AssertionError("Torch not compiled with CUDA enabled") AssertionError: Torch not compiled with CUDA enabled
时间: 2023-10-11 20:16:26 浏览: 188
快速安装Pytorch以及CUDA匹配问题
5星 · 资源好评率100%
这是因为你的 PyTorch 安装包不是专门针对 CPU 的版本,而是同时支持 GPU 和 CPU 的版本。在安装 PyTorch 时,需要确保选择的是针对 CPU 的版本,并且在代码中使用 CPU 相关的函数和方法。
如果你已经安装了 PyTorch 的 CPU 版本,但仍然收到“Torch not compiled with CUDA enabled”的错误,可以尝试在代码开头添加以下语句,以明确使用 CPU:
```python
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
```
这会禁用 CUDA 并强制使用 CPU 运行代码。
阅读全文