Windows安裝CUDA和CUDNN運行環境

由於很多同學需要使用Windows系統進行GPU上的計算，比如運行TensorFlow或者Pytorch等，在Windows上正確安裝CUDA和CUDNN則成了一個關鍵的問題。與在Linux上安裝的整體流程類似，但是還有一些細節要注意，本文以Windows 10系統為例，進行CUDA環境的安裝，並以TensorFlow驗證安裝的成功。1 安裝最新版Microsoft Visual Studio看到這裡很多人可能會問，就寫個Python程序為什麼要裝vs軟體。其實這是因為，我們需要給我們的計算機安裝Windows系統上包括最新版在內的幾個版本的C++編譯器和運行環境。不同於Linux系統，一個嶄新的Windows系統並不包含這些軟體，所以很多安裝TensorFlow GPU版的同學發現安裝之後，運行時報錯：「ImportError: DLL loadfailed: 找不到指定的程序」。我們從TensorFlow官方的文檔 ( https://www.tensorflow.org/install/source_windows) 中就可以看到對編譯器版本的要求這一點：

而這個問題正是網上很多教程不曾寫到的，也是最令人困擾的一點。

我們需要打開微軟VisualStudio的官方網站：

https://visualstudio.microsoft.com/zh-hans/  

下載並安裝最新版VisualStudio，安裝時必須勾選「使用C++的桌面開發」，其他選項可根據自己實際需要勾選。

2 查看本機的CUDA驅動適配版本在桌面右鍵「NVIDIA控制面板」，點擊幫助->系統信息->組件。

在打開的窗口中，我們可以看到本機當前最高支持的CUDA版本是10.2。如果你升級了驅動，將來也可能會支持更高版本。3 下載CUDA和cuDNNCUDA下載頁面：https://developer.nvidia.com/cuda-downloadshttps://developer.nvidia.com/cuda-toolkit-archive

下載cuDNN時必須選擇和你安裝的CUDA匹配的版本，下載頁面：https://developer.nvidia.com/rdp/cudnn-download下載cuDNN是需要登錄英偉達開發者帳戶的，如果沒有的話，需要註冊一個並填寫問卷，很簡單。註冊並登錄後，即可打開如下頁面，選擇對應的文件並下載。

4 安裝CUDA和cuDNN安裝CUDA時很簡單，跟我們平時裝一個軟體一樣，這裡不再贅述。CUDA安裝完成後，打開powershell，執行nvcc -V ，成功的話會返回cuda版本號。安裝cuDNN首先需要解壓cuDNN壓縮包，可以看到有bin、include、lib目錄。

打開 「C:\Program Files\NVIDIAGPU Computing Toolkit\CUDA」 目錄，找到你安裝的版本目錄，打開，找到bin、include、lib目錄，將cuDNN壓縮包內對應目錄下的文件分別複製到bin、include、lib目錄。如果是新版Windows 10系統，可以直接複製三個目錄到對應路徑下，兩處的目錄會自動合併，不會出現目錄級的覆蓋，最多只會覆蓋同名文件。如果是其他版本系統，為了以防萬一，最好還是手動一一複製到對應路徑下。5 添加環境變量我們需要在系統環境變量的Path項下添加幾個路徑，點擊 編輯 -- > 新建、瀏覽，選擇剛才的安裝路徑。

需要添加的默認的安裝路徑如下：

C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v10.2C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64
如果有安裝到別處，請參考這兩個路徑來填寫。
6 檢查安裝結果在任意路徑下打開powershell終端，比如在桌面處按住Shift鍵，滑鼠右鍵，選擇「在此處打開Powershell窗口」。然後輸入：「nvidia-smi」，如果出現類似如下信息，則表明CUDA安裝成功。
然後我們運行python，並」import tensorflow as tf」，輸入」tf.test.is_gpu_available()」，然後回車，如果出現」True」字樣，則說明完全安裝成功。

科技力量是中國抗疫制勝的法寶
基於動態軟體體系結構的插件機制原理(C語言實現)
ASRT開源語音識別項目開始兼容TensorFlow 2.X
[論文翻譯]SpecAugment：一種用於自動語音識別的簡單數據增強方法
手把手教你從零開始搭建個人博客


博客網址：https://blog.ailemon.me

點擊下方「閱讀原文」即可閱讀博客原文，並參與評論和互動。
點在看，讓檸檬菌知道你在看~
                

Long-press QR code to transfer me a reward

As required by Apple's new policy, the Reward feature has been disabled on Weixin for iOS. You can still reward an Official Account by transferring money via QR code.