1 环境配置:Centos7服务器已经安装好anaconda3,并利用anaconda安装TensorFlow(本例中版本为CUDA 8.0.44
CUDNN 5.1.10;Python 3.6.7 :: Anaconda custom (64-bit);TensorFlow为1.12.0);安装后可以利用如下查看版本和检测是否安装正确:
cat /usr/local/cuda/version.txt
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
conda -V
python -V
>>> import tensorflow as tf
>>> tf.__version__
2 Python代码:下载github测试代码,或者自己写一个train.py
git clone https://github.com/gamersover/tensorflow_basic_tutorial
已经写好的minist_Tensorboard.py百度网盘下载,提取密码(8rcr); 这里方便起见用运行自己写的这个minist_Tensorboard.py网络进行展示。建立一个文件夹DemoMinistNN存放下载的minist_Tensorboard.py。
3 运行train.py,报错。手动下载MINIST数据并导入。
在Xshell终端cd 到DemoMinistNN文件夹下,直接运