win10 下GeForce 940MX + CUDA10.0 +pytorch环境配置

前言：

      笔记本显卡是硬伤，为了验证GPU上跑的代码的正确性。就想着利用下笔记本上的显卡：

我的显卡配置：

具体步骤：

一、确定自己的显卡是否支持CUDA

  打开英伟达官网找到 CUDA-Enabled GeForce Products 寻找有没有自己对应的显卡型号，如果有才能下载。

二、安装CUDA

进入CUDA下载链接，根据电脑的属性点击。下载完成后点击安装即可。我下载的是10

三、安装cuDNN

打开cuDNN官网，下载， CUDA10.0对应的cuDNN版本我下载的是cuDNN7.4。

下载完之后将压缩包里面的文件夹“cuda”解压，文件夹里边有有一个“bin”文件，里面存有“cudnn64_5.dll”文件。

加入存有该文件的路径地址“D:\cuda&cuDnn\cuda\bin”到Path环境变量中

最后点击确认即可。

四、安装pytorch 稳定版本

 进入pytorch官网。我使用的是pip安装命令。

五、验证

进入anaconda Prompt ，输入：

>>> python
>>> import torch
>>>  torch.cuda.is_available()

如果显示True，则正常。

参考链接： http://ddrv.cn/a/586382