环境配置(近期实测)——Ubuntu16.04+CUDA9.0+tensorflow-gpu填坑记
近几年深度学习在物体检测方面出现了许多基于不同框架的网络模型,不同模型需要不同的版本的Python、TensorFlow、Keras、CUDA、cuDNN以及操作系统。不得不说,要把经典物体检测网络的源码都跑通,单配置环境就要浪费很多时间,因为目前兼容这些经典网络的框架和环境还很少。新版的TensorFlow在models的 objection-detection 模块中包含了fast-rcnn、rfcn、SSD几种网络,不过开发环境是基于Ubuntu的,因此我对该环境进行了配置。 我的运行环境为: Ubuntu16.04LTS CUDA9.0 cuDNN7.0.5 AMD Ryzen 1600 NVIDIA GTX1070 1.Ubuntu安装环境选择 Ubuntu版本很多,首选LTS版本。现在已经有了18.04LTS,装上后界面丰富好看了,不过在运行Python和TensorFlow环境下的代码时出现了一些未知问题,有关排故的信息也比较少,所以我重新安装了16.04LTS。安装时可先使用 深度U盘启动工具 (我使用的是UEFI版)制作Ubuntu启动盘,然后在电脑安装,具体过程百度即可。我以前使用的是双系统安装,但时间长了以后Ubuntu系统会出问题,有时需要进入grub模式进行修复,操作不当会导致无法进入Windows系统。所以后来我就把两个系统安装在两个硬盘上