FCN在windows下利用sift--flow數據集訓練網絡模型所遇到的問題

參考配置鏈接:https://www.cnblogs.com/xuanxufeng/p/6243342.html

1.運行python solve.py報錯:

解決:參考https://blog.csdn.net/u012198575/article/details/98957901

caffe的模式爲cpu-only,需要重新設置模式並重新編譯caffe

修改build_win.cmd文件

將76行改爲CPU_ONLY=0

添加一個-DCUDNN_ROOT=你cuDNN的路徑(先安裝cudnn)上面有註釋,必須是cudnn8.0

重新編譯caffe,不報錯

 

2.cpu模式訓練fcn,修改solve.py文件

 

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