1. 程式人生 > >Ubuntu下配置TensorFlow-Gpu版,其對應CUDA、cuDNN版本

Ubuntu下配置TensorFlow-Gpu版,其對應CUDA、cuDNN版本

TensorFlow官網安裝指導介面提示“執行支援 GPU 的 TensorFlow 所需滿足的 NVIDIA 要求”

檢視機器CUDA、cuDNN版本命令分別是:

  1. cat /usr/local/cuda/version.txt
  2. cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

 例如得到如下資訊:

CUDA Version 8.0.44

#define CUDNN_MAJOR      5
#define CUDNN_MINOR      1
#define CUDNN_PATCHLEVEL 5

 這表示CUDA為8.0版,cuDNN為5.1.5

官網給出了經過測試的原始碼配置資訊:

以linux為例:

 可以看到1.0~1.2是我們可以使用的GPU版。