AssersionError:火炬未启用CUDA编译

时间:2020-08-16 17:49:21

标签: python pytorch torch

我要运行此repo。我安装了该项目所需的所有东西。

我有Windows 8.1操作系统,似乎没有NVIDIA GPU(从设备管理器:显示适配器-AMD Radeon HD 7660G + 7670M Dual Graphics和AMD Radeon HD 7670M)。

我用Pytorch web-site

上显示的命令安装了割炬
pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

但是当我运行项目时,我收到错误消息- AssersionError:Tourch未在启用CUDA的情况下编译

然后我尝试在启用CUDA的情况下安装割炬。

pip install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

但是当我运行项目时,我收到错误消息- AssersionError:您的系统上未找到NVIDIA驱动程序。请检查您是否具有NVIDIA GPU并从https://www.nvidia.com/Download/index.aspx 安装了驱动程序。

请帮助我解决问题并正确运行项目。

1 个答案:

答案 0 :(得分:0)

我已经解决了这个问题。他们使用的源代码中存在一个问题

opt.device

在main.py和violin_dataset.py中。但这被声明为

opt.device = torch.device('cuda:0')
即使您没有cuda支持,也可以在config.py中使用

。 所以我将其更改为

opt.device = torch.device('cpu')

现在一切正常。