- 首先查看自己的cuda是否可用
torch.cuda.is_available()
这里我的cuda是不适配torch的,所以需要重新安装适配的torch
- 查看自己的cuda版本
方法1
方法2
在cmd处输入nvidia-smi
这样可以找到的自己的CUDA版本 - 安装符合自己版本的pytorch
进入pytorch官网https://pytorch.org/get-started/locally/安装自己需要的pytorch
这里都是最新版的Pytorch,如何没有你需要的版本,可以安装之前版本的pytorch
因为我的cuda版本为11.7,这里我就安