チェック失敗：エラー== cudaSuccess（3対0）初期化エラー***不良スタックトレースを確認してください：**

Question

私は非常にcudaとcaffeです。私のコードでは、1つのテキストライン検出に1つのcaffeモデルを使用し、chracAter認識にもう1つのcaffeモデルを使用しています。テキストラインを検出した後、セグメンテーションと認識のためにすべてのテキストラインを並行して処理しています。しかし、認識をしている間、次のようなエラーが表示されます。

I0503 14:40:41.661458 3996 net.cpp:436] Input 0 -> data 
I0503 14:40:41.661509 3996 layer_factory.hpp:76] Creating layer scale 
I0503 14:40:41.661527 3996 net.cpp:111] Creating Layer scale 
I0503 14:40:41.661536 3996 net.cpp:478] scale <- data 
I0503 14:40:41.661545 3996 net.cpp:434] scale -> scaled 
I0503 14:40:41.661563 3996 net.cpp:156] Setting up scale 
I0503 14:40:41.661576 3996 net.cpp:164] Top shape: 1 1 20 20 (400) 
I0503 14:40:41.661583 3996 layer_factory.hpp:76] Creating layer conv1 
I0503 14:40:41.661597 3996 net.cpp:111] Creating Layer conv1 
I0503 14:40:41.661605 3996 net.cpp:478] conv1 <- scaled 
I0503 14:40:41.661615 3996 net.cpp:434] conv1 -> conv1 
F0503 14:40:41.661710 3996 syncedmem.hpp:19] Check failed: error == cudaSuccess (3 vs. 0) initialization error 
*** Check failure stack trace: *** 

どうすればこの問題を解決できますか？

Answer 1

この問題も発生しました。今、あなたを助ける助言がいくつかあります。

1. 各スレッドの全体のcaffeネットを初期化します。

2. Caffe::mode_これを制御する変数はスレッドローカルなので、Caffe関数を実行する前に各スレッドでcaffe.set_mode_gpu()を呼び出していることを確認してください。

3. マルチプロセッシングモジュールの代わりにスレッドモジュールを使用します。