1、nvidia官网：https://www.nvidia.cn/
2、找到 驱动程序 -> 所有 NVIDIA驱动程序 找到对应显卡的最新驱动
3、根据自己电脑的显卡型号 查找驱动，如 GTX1050
4、找到后，下载安装即可
问题：NVIDIA-SMI has failed because you are not:
a) running as an administrator or
b) there is not at least on .........
原因：因Win10 系统更新导致的新的内核 与 nvidia驱动不匹配，导致的 GPU不工作
重装驱动后问题解决：
在 命令提示符窗口 cd C:\Program Files\NVIDIA Corporation\NVSMI ,然后输入 nvidai-smi 会输出显卡与驱动信息，若驱动程序不匹配，则会报错信息。
cuda 和 cudnn安装
5、cuda 下载地址：cuda
根据自己笔记本 选择相应配置：
6、cuDNN下载地址： https://developer.nvidia.com/rdp/cudnn-archive
根据步骤5下载的 cuda 版本下载对应的 cudnn即可
7、下载安装对应GPU版本的 pytorch 即可
pytorch 下载地址：https://pytorch.org/get-started/locally/
Reference: