win10+GeForce 940mx安装cuda

起因是在python3里执行下述代码，发现是false，于是想想怎么解决。

import torch
torch.cuda.is_available()

首先搜到在windows中安装cuda的指引，发现主要步骤为：

1. 配平显卡驱动、cuda版本、pytorch版本。

我是pytorch 1.11.0装好了才发现这个问题，根据前车之鉴，我选择把显卡驱动升级为最新版本512.95，并开始准备安装1.11.0的cuda。

注意

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

这里的cudatoolkit和NVIDIA的不是同一个东西，两者都需要安装。

2. 安装cuda

参考官方安装教程以及这个博客的第三步和第四步，下载win对应版本的Installer，全部安装后验证PASS。

注意：安装包2.5G，从安装解压到结束至少需要13G的空间，最终安装好的大小为6G左右

3. 安装cuDNN

参考官方安装教程，找到对应的版本下载后解压。

4. 验证

5. GPU不起作用

可能遇到的问题：

pytorch和cudatoolkit版本严格对应

参考文章：

cuda与显卡驱动版本对照

全流程安装步骤