代码链接:
https://github.com/endernewton/tf-faster-rcnn
1. 前期准备
- 安装虚拟环境
创建一个tf-faster-rcnn的python2.7的虚拟环境
conda create –n tf-faster-rcnn python=2.7
进入此虚拟环境
source activate tf-faster-rcnn
备注:
a. 如果需要退出此虚拟系统,执行以下命令
source deactivate
b. 在虚拟环境中安装的东西会放在此根目录下/home/USERNAME/.conda/envs/tf-faster-rcnn(创建的虚拟环境名字)
查看cuda的版本
cat /usr/local/cuda/version.txt
查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grepCUDNN_MAJOR -A 2
根据cuda 和 cudnn版本,到tensorflow 的官网上查对应的TF版本。我的cuda是8.0,cudnn是5.1的,所以要安装1.2.0
https://www.tensorflow.org/install/install_sources#common_installation_problems
2. 环境配置