iT邦幫忙

0

[Python] Tensorflow with GPU

使用Python 做機器學習/深度學習時常使用的Tensorflow
還有分成CPU版本與GPU版本
如果要安裝GPU版本,還需要安裝 nvida Cuda & cnDNN

  • Cuda(Compute Unified Device Architecture)是NVIDIA所推出的一種整合並行計算框架
  • Cudnn(CUDA Deep Neural Network library)是NVIDIA打造的針對深度學習網路的加速library

安裝 Cuda & cnDNN 前需要先確認tensorflow.org支援對應的版本

Software requirements

從TS官網,確認需要cuda9.0 / cuDNN7.2+

下載安裝Cuda,這部份很簡單,一直按下一步下一步下一步下一步就好了
要注意下載對應的版本,這次需要的是9.0版
Cuda

緊接著是cuDNN,注意7.2 fow windows
hcuDNN

解壓縮後將bin\DLL位置加入系統環境變數

建立CONDA 工作環境 : conda create --name trensorflow-gpu python=3.6 anaconda nb_conda

成功
https://ithelp.ithome.com.tw/upload/images/20190212/20115086dW703GriDI.png


尚未有邦友留言

立即登入留言