nvidia驱动、cuda和GPU版本pytorch安装

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:

GPU版pytorch安装