- 首先查看自己的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,这里我就安装的pytorch1.13.1版本的
pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
安装成功
查看这时cuda是否可用
这里显示为true,完美解决,撒花
注:在这个过程中,我还尝试了安装torch2.0.0和2.0.1版本,但是都不可用,显示的是cpu版本的,我在查阅问题原因,说的是cuda11.7不接受2.0以上的版本,建议各位安装2.0以下的版本。