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