Pytorch GPU版本安装

一、背景

       记录一下安装Pytorch GPU版本过程。

       由于手残，卸载了电脑上的显卡驱动，现在我连显卡类型是啥都不知道了。

       总体思路：安装显卡驱动->安装cuda->安装pytorch库

二、安装显卡驱动

2.1 查看本地显卡型号

        通过「DirectX 诊断工具」查看显卡信息，以便下载驱动程序。

  键盘win+r  输入 dxdiag 进入诊断工具界面

2.2 下载显卡驱动程序

        进入人工智能计算领域的领导者 | NVIDIA英伟达官网，点击驱动程序。

        根据自己电脑配置信息进行选择，然后点击搜索->