node3 装tensorflow

1.cudnn 安装

师兄没有在node3装cudnn,于是重新装了一遍.版本查看:

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

现在装的是cuda8.0,cudnn6.0

安装过程

官网下载 cudnn-8.0-linux-x64-v6.0.tgz

解压到名为cuda的文件夹

tar zxvf cudnn-8.0-linux-x64-v6.0_1.tgz cuda

复制文件

cd cuda/include 
sudo cp *.h /usr/local/cuda/include/
cd cuda/lib64 
sudo cp libcudnn* /usr/local/cuda/lib64/

最后终端运行

chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* 

2.tensorflow安装

sudo apt-get install libcupti-dev
conda create -n tensorflow pip python=2.7
source activate tensorflow
pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.3.0-cp27-none-linux_x86_64.whl

完成