1. 程式人生 > >CUDA+CUDNN+Tensorflow安裝

CUDA+CUDNN+Tensorflow安裝

一、安裝cuda

首先下載cuda打deb檔案https://developer.nvidia.com/cuda-80-ga2-download-archive

選擇相應的系統版本和檔案型別(此處我們選擇local的deb型別檔案)

然後執行以下三條命令

1、sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb

2、sudo apt-get update

3、sudo apt-get install cuda

重新啟動,然後輸入invidia-smi驗證,有輸出則證明安裝成功。

二、安裝cudnn

首先下載檔案,連結https://developer.nvidia.com/rdp/cudnn-download

選擇Download cuDNN v6.0 (April 27, 2017), for CUDA 8.0下的cuDNN v6.0 Library for Linux版本

直接的下載連結為https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v6/prod/8.0_20170307/cudnn-8.0-linux-ppc64le-v6.0-tgz

下載完成後在命令列中輸入以下四條命令

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/

sudo chmod a+r /usr/local/cuda/include/cudnn.h

sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

三、安裝Tensorflow-gpu

首先安裝pip

sudo apt-get install python-pip
 

然後使用pip安裝tensorflow,此處注意。我們要安裝的是tensorflow的gpu版本,而且,cuda、cundnn、tensorflow和python的版本需要適當配合才能正確執行,也就是說不可以隨意更換版本,我們這裡要安裝tensorflow-gpu的1.4版本,他和cuda8、cudnn6、python2或者3都可以很好的配合。所以執行以下命令
pip install tensorflow-gpu==1.4
 
如果中途遇到許可權或者pip更新版本問題,根據錯誤提示使用sudo許可權,或者升級pip版本即可
完成之後,
在命令列下輸入python進入python

import tensorflow as tf 嘗試,

嘗試執行一個程式,然後看大會輸出你的顯示卡資訊就證明ok啦