1.先查询自己安装的pytorch版本
我这个是Anaconda 需要到对应的虚拟环境中查看
python环境可以直接通过win+F4(cmd) pip list 查看
看有没 +cu102 这个 后缀 有 那就可能是版本兼容问题,没有那就是下载到cpu版了
2.安装对应版本https://download.pytorch.org/whl/torch_stable.html
注意不要下载到cpu开头的文件
下载自己对应的cuda文件
cu102 cuda
cp37 python3.7
linux系统
win系统
引用文档
Lambda_c
GPU版本的pytorch安装(一次成功)_pip install torch gpu_Lambda_c的博客-CSDN博客