1. 程式人生 > >cuda、cudnn、tensorflow以及nvidia GPU驅動安裝

cuda、cudnn、tensorflow以及nvidia GPU驅動安裝

有一臺伺服器,在使用nvidia-smi命令是,提示:

NVIDIA-SMI has failed because it couldn’t communicate with the NVIDIA driver. Make sure that the latest NVIDIA driver is installed and running.

解決方法(需有cuda安裝經驗):
1)先解除安裝nvidia驅動以及cuda:
2)安裝cuda,根據自己的需要安裝即可,在安裝cuda時,會提示是否安裝驅動,選擇安裝驅動,即可修復,如果採用runfile檔案安裝失敗,那麼就換成rpm檔案(centos)或者deb(ubuntu、debian)進行安裝。

相應環境變數設定(以cuda8.0為例,其他只需修改版本號即可):

export CUDA_HOME=/usr/local/cuda-8.0
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$CUDA_HOME/lib:$PATHH

tensorflow版本與cuda cuDNN版本對應使用

tensorflow-gpu v1.9.0 | cuda9.0 |  cuDNN7.1.4可行  | 備註:7.0.4/ 7.0.5/ 7.1.2不明確
tensorflow-gpu v1.8.0 | cuda9.0 |  cuDNN  不明確 | 備註:7.0.4/ 7.0.5/ 7.1.2/ 7.1.4
tensorflow-gpu v1.7.0 | cuda9.0 | cuDNN 不明確 | 備註:7.0.4/ 7.0.5/ 7.1.2/ 7.1.4 tensorflow-gpu v1.6.0 | cuda9.0 | cuDNN 不明確 | 備註:7.0.4/ 7.0.5/ 7.1.2/ 7.1.4 tensorflow-gpu v1.5.0 | cuda9.0 | cuDNN 不明確 | 備註:7.0.4/ 7.0.5/ 7.1.2/ 7.1.4 tensorflow-gpu v1.4.0 | cuda8.0 | cuDNN 6.0 | 備註:6.0正常使用, 7.0.5不能用,5.1未知 tensorflow-gpu v1.3.0
| cuda8.0 | cuDNN 6.0 | 備註:6.0正常使用, 7.0.5不能用,5.1未知 tensorflow-gpu v1.2.0 | cuda8.0 | cuDNN 5.1 | 備註:5.1正常使用, 6.0/ 7.0.5 未知 tensorflow-gpu v1.1.0 | cuda8.0 | cuDNN 5.1 | 備註:5.1正常使用, 6.0/ 7.0.5 未知

檢測tensorflow是否使用gpu

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))