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