AssertionError: Torch not compiled with CUDA enable
时间: 2023-11-10 11:04:35 浏览: 104
快速安装Pytorch以及CUDA匹配问题
5星 · 资源好评率100%
这个错误提示表明你的PyTorch没有启用CUDA,也就是没有安装GPU版本的PyTorch或者你的环境没有正确配置。你需要安装GPU版本的PyTorch并且确保你的CUDA驱动和CUDA工具包版本与PyTorch要求的版本匹配。
如果你已经安装了GPU版本的PyTorch,可以尝试重新安装或者更新CUDA驱动和CUDA工具包。
阅读全文