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。

2023年4月24日：补充具体步骤

1. 在Previous PyTorch Versions | PyTorch网址中找自己已经装了的pytorch版本号（如1.8.0）

注意，如果之前已经安装了（像我就错装了cpu版本的torch），必须要先卸载（最好pip和conda都uninstall一下），不然就会遇到问题：Key already registered with the same priority: GroupSpatialSoftmax，重装解决一切问题。

 

按照提示安装到conda环境里：

安好后验证一下：

至于nvcc -c的使用：

注意

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与显卡驱动版本对照

全流程安装步骤