CUDA是NVIDIA GPU用於平行運算的程式架構,我們可以利用CUDA進行深度學習的建模,由於目前大多GPU的運算速度是高於CPU,所以利用GPU加速深度學習模型運算是目前最主流的方法。因此,確認過電腦的GPU是支援CUDA運算架構之後,我們就開始來安裝CUDA。
TensorFlow (gpu)
CUDA
cuDNN
更新時間
(最新) 1.7
9.1
7.1.2
2018.4 (最新)
1.5
9.0
7(.0.4/5)
2018.2
1.3
8.0
6.0
2017.11
1.0
5.1
2017.8
請下載目前CUDA最新9.1版(2018.4 更新),並請確認是否支援以下作業系統:
Table 1. Windows Operating System Support in CUDA 9.1
Native x86_64
Cross (x86_32 on x86_64)
Windows 10
YES
Windows 8.1
Windows 7
Windows Server 2016
NO
Windows Server 2012 R2
安裝CUDA之前,必須先安裝Windows Visual Studio,請確認安裝的版本。
Table 2. Windows Compiler Support in CUDA 9.1
Compiler
IDE
Visual C++ 15.0
Visual Studio 2017
Visual C++ 14.0
Visual Studio 2015
Visual Studio Community 2015
Visual C++ 12.0
Visual Studio 2013
Visual C++ 11.0
Visual Studio 2012
Visual C++ 10.0 DEPRECATED
Visual Studio 2010
資料來源:http://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html#axzz4oJ4It0ksarrow-up-right
1.安裝 Visual C++ Redistributable for Visual Studio 2015arrow-up-right
2.安裝 CUDA 9.1arrow-up-right
選擇對應的作業系統環境,安裝CUDA和Patch。(安裝非最新版CUDA請至https://developer.nvidia.com/cuda-toolkit-archivearrow-up-right 下載)
補充:舊版CUDA須從歷史安裝頁面下載
nvidia-smi -l 1
3.安裝 cuDNNarrow-up-right
https://developer.nvidia.com/rdp/cudnn-archive
continually give you the gpu usage info, with in refresh interval of 1 second
Last updated 7 years ago