window10编译tensorflow-fasterrcnn训练环境
1、显卡驱动:quadroK620
2、CUDA8.0
查看cuda版本: 命令行 nvcc -V
3、cudnn v6.0
4、Anaconda3 4.2.0 (-----对应的python3.5)
5、tensorflow-gpu 版本 tensorflow-gpu-1,3,0.0rc0
另外需要的package有:
protobuf-3.5.1
tensorflow-tensorboard-0.1.8
html5lib-0.9999999
markdown>=2.6.8
bleach-1.5.0
升级pip
python -m pip install --upgrade pip
pip版本:pip-9.0.1
6、安装cython, python-opencv, easydict
7、faster-rcnn: https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3.5
FOLLOW README
Pretrain-model: https://github.com/tensorflow/models/tree/master/research/slim#pre-trained-models
8、出现错误:
(1)numpy版本升级
(2)opencv与python版本对应问题(原本VS版本是2013, 可能要换成2015)
这个问题可以参考:http://blog.****.net/yingyueer247/article/details/76422848
找对应版本的opencv: https://www.lfd.uci.edu/~gohlke/pythonlibs/#opencv
后来可以了 但是还是有问题 这个显卡的显存是2G 可能是显存不够
batch size 改成1了。。。。
换台电脑试试吧
------------------------------------------------------------------------------------------------------------------------------------------------------------------------------分割线
换了笔记本 显卡960M
同样的步骤 在跑train.py的时候报错
ImportError: No module named '_pywrap_tensorlfow_internal
检查了环境变量 发现cuda\bin没有在环境变量path里
.\CUDA\bin 要加入环境变量path
在装cuda时 就有提示了。 需要安装后自己改下路径(把路径缩短),再手动加入环境变量
用了自己的数据试试 batch size给的2 可以了