錯誤以便更快地-R-CNN培訓模式在Tensorflow，但Caffe實施

我使用它實現所謂的更快R-CNN一個深度學習模型tensorflow以下GitHub的倉庫，我想：錯誤以便更快地-R-CNN培訓模式在Tensorflow，但Caffe實施

https://github.com/smallcorgi/Faster-RCNN_TF

然後，我運行命令（我想只能用CPU來運行）：

python ./tools/demo.py --model VGGnet_fast_rcnn_iter_70000.ckpt

我得到的錯誤：

Traceback (most recent call last): 
    File "./tools/demo.py", line 11, in <module> 
    from networks.factory import get_network 
    File "/home/brm17/Faster-RCNN_TF/tools/../lib/networks/__init__.py", line 8, in <module> 
    from .VGGnet_train import VGGnet_train 
    File "/home/brm17/Faster-RCNN_TF/tools/../lib/networks/VGGnet_train.py", line 2, in <module> 
    from networks.network import Network 
    File "/home/brm17/Faster-RCNN_TF/tools/../lib/networks/network.py", line 3, in <module> 
    import roi_pooling_layer.roi_pooling_op as roi_pool_op 
    File "/home/brm17/Faster-RCNN_TF/tools/../lib/roi_pooling_layer/roi_pooling_op.py", line 5, in <module> 
    _roi_pooling_module = tf.load_op_library(filename) 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/load_library.py", line 64, in load_op_library 
    None, None, error_msg, error_code) 
tensorflow.python.framework.errors_impl.NotFoundError: /home/brm17/Faster-RCNN_TF/tools/../lib/roi_pooling_layer/roi_pooling.so: undefined symbol: _Z22ROIPoolBackwardLaucherPKffiiiiiiiS0_PfPKiRKN5Eigen9GpuDeviceE

是什麼問題，以及如何解決這個問題？

來源

2017-05-08 Rahim Kasmi

問題是，您沒有按照存儲庫中的說明。 –

歡迎來到StackOverflow。請刪除標記'caffe'，因爲它不相關。 – lnman

首先，您需要安裝cuda（7.5或8.0）（link to install cuda-8.0）。然後，你需要在你的.bashrc文件（位於在您的家）去，並添加以下兩行：

出口LD_LIBRARY_PATH =在/ usr /本地/ CUDA/lib目錄

出口PATH = $ PATH：在/ usr /本地/ CUDA/bin中

一旦.bashrc文件保存，運行以下命令：

源的.bashrc

不要忘記，以驗證兩個變量是否存在（簡單地通過輸入$ LD_LIBRARY_PATH和$ PA TH）。如果他們不這樣做，請轉到路徑以更改文件夾的名稱（例如，我必須通過第一個變量中的「lib64」更改「lib」）。

然後，它應該工作:-)

來源

2017-08-14 15:49:29

錯誤以便更快地-R-CNN培訓模式在Tensorflow，但Caffe實施

回答

相關問題