1.输入如下命令
pythonimport torchtorch.__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
再验证一下是否可以