1. 运行环境介绍
- NVIDIA GTX 1070
- Ubuntu 16.04 x64
- CUDA 8.0.61
- cuDNN 5.1
- Python 3.4
- TensorFlow 1.2.0
- PyGame 1.9.4
- OpenCV 3.1.0 for Python 3.4(opencv-3.1.0-np111py34_1.tar.bz2)
- OpenCV-Python for Python 3.4(opencv3-3.1.0-py34_0.tar.bz2)
2. 准备
S1. 假定已经安装好合适版本的CUDA、cuDNN、Python和TensorFlow。本笔记省略相关步骤介绍,具体可以参考以下两篇文章: