AssertionError: Torch not compiled with CUDA enabled如何解决
时间: 2023-11-12 16:59:03 浏览: 72
解决AssertionError Torch not compiled with CUDA enabled.docx
这个错误通常是因为你的PyTorch没有安装CUDA或者CUDA版本不匹配导致的。你可以尝试以下几种方法解决这个问题:
1. 确认你的PyTorch版本是否支持CUDA,可以通过以下命令查看:
```
import torch
print(torch.version.cuda)
```
如果输出为空或者为None,则说明你的PyTorch没有安装CUDA。
2. 确认你的CUDA版本是否与PyTorch版本匹配,可以通过以下命令查看:
```
nvcc --version
```
然后在PyTorch官网上查看对应版本的PyTorch支持的CUDA版本。
3. 如果你的CUDA版本与PyTorch版本匹配,但是仍然出现这个错误,可以尝试重新安装PyTorch并指定CUDA版本,例如:
```
pip install torch==1.7.0+cu101 -f https://download.pytorch.org/whl/cu101/torch_stable.html
```
其中cu101表示CUDA 10.1版本,你可以根据自己的CUDA版本选择对应的版本号。
阅读全文