Merge pull request PaddlePaddle#48 from wangye707/wangye_fix_DeviceGe…

…tHandle_bug 修复gpu返回str问题
Zeref996 · Jul 23, 2021 · 2317b26 · 2317b26
2 parents e482ab4 + eb664e1
commit 2317b26
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/inference/python_api_test/test_case/infer_test.py b/inference/python_api_test/test_case/infer_test.py
@@ -114,7 +114,7 @@ def disable_gpu_test(self, input_data_dict: dict, repeat=20):
 
         cuda_visible_devices = os.environ.get("CUDA_VISIBLE_DEVICES")
         if cuda_visible_devices:
-            cuda_visible_device = cuda_visible_devices.split(",")[0]
+            cuda_visible_device = int(cuda_visible_devices.split(",")[0])
         else:
             cuda_visible_device = 0
 

diff --git a/inference/python_api_test/test_class_model/run.sh b/inference/python_api_test/test_class_model/run.sh
@@ -1,4 +1,3 @@
-python -m pip install pytest
 export FLAGS_call_stack_level=2
 cases=`find . -name "test*.py" | sort`
 ignore=""
@@ -11,7 +10,7 @@ do
     if [[ ${ignore} =~ ${file##*/} ]]; then
         echo "跳过"
     else
-        python3.7 -m pytest ${file}
+        python -m pytest ${file}
         if [ $? -ne 0 ]; then
             echo ${file} >> result.txt
             bug=`expr ${bug} + 1`