校验pytorch是否支持显卡GPU 不支持卸载并安装支持版本

1.输入如下命令

python

import torch

torch.__version__

torch.cuda.is_available() // 输出False 就是不支持

如下图

2.可以看到我电脑目前是不支持的 我们现在开始卸载

 exit() //先退出
 
 pip uninstall torch //开始卸载

这就卸载完成了

3.我们开始安装

nvidia-smi.exe  //运行这个地 查看你CUDA版本号

安装pytorch 的版本号不得高于该版本号！

打开 pytorch 官网 下载

复制这条命令 去下载

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126

再验证一下是否可以