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.csdn.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  可以了




猜你喜欢

转载自blog.csdn.net/summermaoz/article/details/78963254