集成显卡运行程序时报错 raise AssertionError("Torch not compiled with CUDA en
时间: 2024-04-27 17:18:48 浏览: 125
快速安装Pytorch以及CUDA匹配问题
5星 · 资源好评率100%
这个错误是由于在使用集成显卡运行程序时,程序需要使用CUDA加速,但是当前安装的PyTorch没有编译CUDA。您需要重新安装PyTorch并确保它被编译了CUDA。
以下是在Windows系统上安装PyTorch的命令:
```
pip install torch===1.7.1 torchvision===0.8.2 torchaudio===0.7.2 -f https://download.pytorch.org/whl/cu101/torch_stable.html
```
请注意,此命令安装的是PyTorch 1.7.1,如果您需要其他版本,请查看PyTorch官方网站上的支持矩阵。
阅读全文