1.1 关于CUDA:
tensorflow-gpu 1.5版本及以上要求CUDA版本为9.0
查看本机CUDA版本方法:
cat /usr/local/cuda/version.txt
1
输出:
CUDA Version 8.0.61`
1
1.2 关于cuDNN:
tensorflow-gpu 1.3及以上版本要求cudnn版本为V6及以上
查看本机cuDNN版本方法:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
1
输出:
#define CUDNN_MAJOR 5
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 5
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include "driver_types.h"
1
2
3
4
5
6
7
1.3 关于tensorflow
由上可以看出本机看装了CUDA 8和cuDNN V5,根据这两个版本,选择tensorflow的版本为1.2,使用pip来安装tensorflow:
sudo pip install tensorflow-gpu==1.2
1
如果之前安装了高版本的tensorflow,那么要通过pip来全部删除:
sudo pip uninstall tensorflow
1
想