已解决：Torch not compiled with CUDA enabled

最近在 Linux 服务器上跑项目，要用到 Python 3.7 + Torch 1.5.0 + Cuda + Transformer。运行过程中出现报错 “Torch not compiled with CUDA enabled”，经过思考认为，应该是 Torch 和 Cuda 版本不匹配导致的，最后成功解决。

解决思路如下：

1. 检查 Nvidia & Cuda 版本，使用 Linux 指令 ：

nvidia-smi

2. 登录 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.html

5. 希望对你有帮助！好用请点赞/ 评论 / 关注，期待与你一起交流，共同进步！

参考：AssertionError: Torch not compiled with CUDA enabled · Issue #30664 · pytorch/pytorch · GitHub