pert_image, pert = deepfool(image, model)为什么报错AssertionError: Torch not compiled with CUDA enabled
时间: 2023-10-11 13:29:34 浏览: 130
项目管理 PERT估算方法的详细介绍和例子
这个错误是因为你的 PyTorch 没有启用 CUDA,也就是没有安装支持 CUDA 的 PyTorch 版本。要解决这个问题,你需要在代码中设置正确的 CUDA 设备。你可以使用以下命令安装支持 CUDA 的 PyTorch:
```
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
其中,cu111 表示 CUDA 11.1 版本,如果你安装的是其他版本的 CUDA,可以根据实际情况替换。同时,在代码中,你需要设置正确的 CUDA 设备,例如:
```
import torch
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model.to(device)
# deepfool函数中也需要设置device
pert_image, pert = deepfool(image.to(device), model.to(device))
```
这样就可以把模型移动到正确的 CUDA 设备上运行了。
阅读全文