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




發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章