最近在 Linux 服务器上跑项目，要用到 Python 3.7 + Torch 1.5.0 + Cuda + Transformer。运行过程中出现报错 “Torch not compiled with CUDA enabled”，经过思考认为，应该是 Torch 和 Cuda 版本不匹配导致的，最后成功解决。
解决方法如下：
1. 检查 Nvidia & Cuda 版本，使用 Linux 指令 ：
nvidia-smi2. 登录Nvidia查看已安装 Nvida & Cuda 版本是否适配，若不适配，需重装Cuda；
3. 登录Pytorch找到与 Nvida & Cuda 适配的 Torch 版本，复制下载指令；
4。例如，我的 Linux 服务器是 NVIDIA-SMI 418.43, 则 Cuda 10.1 与之适配。相应地，我如果想安装 Torch 1.5.0 ，则对应的 Linux 指令为
pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html5. 希望对你有帮助！好用请点赞/ 评论 / 关注，期待与你一起交流，共同进步！
参考：AssertionError: Torch not compiled with CUDA enabled · Issue #30664 · pytorch/pytorch · GitHub
版权声明：本文为博主Jenny300原创文章，版权归属原作者，如果侵权，请联系我们删除！
原文链接：https://blog.csdn.net/weixin_47618408/article/details/123216129
