Pytorch错误：Torch not compiled with CUDA enabled

一、问题

今天跑一个项目代码遇到了一个问题，大致是pytorch和CUDA版本不匹配，然后我就降低pytorch版本，结果1.7中网络不包含我所需的函数，我直接pip install pytorch==1.8，这样产生了错误：Torch not compiled with CUDA enabled。

二、解决方法

nvcc -V

检查一下当前CUDA版本

根据自己的CUDA版本下载安装whl文件

参考链接：https://download.pytorch.org/whl/torch_stable.html

分别

pip install torch-1.8.1+cu101-cp38-cp38-linux_x86_64.whl
pip install torchcsprng-0.2.1+cu101-cp38-cp38-linux_x86_64.whl
pip install torchvision-0.9.1+cu101-cp38-cp38-linux_x86_64.whl