Fasterrcnn AssertionError: Torch not compiled with CUDA enabled
时间: 2023-11-07 12:05:48 浏览: 103
Fasterrcnn报错 "AssertionError: Torch not compiled with CUDA enabled" 是由于当前的Torch版本没有启用CUDA编译所致。您可以尝试以下解决方法:
1. 确认CUDA是否已正确安装。您可以使用`nvidia-smi`命令来检查CUDA是否已正确安装,并且您的显卡驱动程序是否与所安装的CUDA版本兼容。
2. 检查您当前安装的Torch版本是否支持CUDA。您可以通过运行以下命令来检查Torch是否启用了CUDA编译:
```python
import torch
print(torch.cuda.is_available())
```
如果返回`False`,则表示您当前安装的Torch版本未启用CUDA编译。您需要安装与您CUDA版本兼容的Torch版本。
3. 如果您的Torch版本不支持CUDA,您可以尝试安装与您的CUDA版本兼容的Torch版本。您可以使用以下命令来安装特定版本的Torch:
```shell
pip install torch==1.7.1+cu102 -f https://download.pytorch.org/whl/torch_stable.html
```
在上述命令中,`1.7.1`是Torch的版本号,`cu102`表示CUDA版本为10.2。根据您的CUDA版本选择合适的Torch版本进行安装。
4. 如果您之前已安装了不支持CUDA的Torch版本,您可以使用以下命令卸载已安装的Torch:
```shell
pip uninstall torch
```
卸载完成后,根据上述步骤重新安装与您的CUDA版本兼容的Torch版本。
阅读全文