ソースを参照

model service

丁云鹏 5 ヶ月 前
コミット
e9a22d0752
1 ファイル変更6 行追加5 行削除
  1. 6 5
      recommend-model-jni/paddle/test.sh

+ 6 - 5
recommend-model-jni/paddle/test.sh

@@ -14,10 +14,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-library_path=../paddle_inference_c
-mkldnn_lib=$library_path"/third_party/install/mkldnn/lib"
-mklml_lib=$library_path"/third_party/install/mklml/lib"
-paddle_inference_lib=$library_path"/paddle/lib"
-export LD_LIBRARY_PATH=$mkldnn_lib:$mklml_lib:$paddle_inference_lib:.
+library_path=../src/main/resources
+mkldnn_lib=$library_path"/lib/mkldnn"
+mklml_lib=$library_path"/lib/mklml"
+onnx_lib=$library_path"/lib/onnx"
+paddle_inference_lib=$library_path"/lib/paddle"
+export LD_LIBRARY_PATH=$onnx_lib:$mkldnn_lib:$mklml_lib:$paddle_inference_lib:.
 javac -cp $CLASSPATH:JavaInference.jar:. ../src/test/java/com/baidu/paddle/inference/test.java
 java -cp $CLASSPATH:JavaInference.jar:../src/test/java:. com.baidu.paddle.inference.test ../src/test/resources/inference.pdmodel ./src/test/resources/inference.pdiparams