From d2ffdf7703f7639761055d4fbb2b1347eeba1f5c Mon Sep 17 00:00:00 2001 From: MRXLT Date: Tue, 7 Apr 2020 02:59:47 +0000 Subject: [PATCH] fix cuda check --- python/paddle_serving_server_gpu/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle_serving_server_gpu/__init__.py b/python/paddle_serving_server_gpu/__init__.py index 4cbfbf96..cfd5eee9 100644 --- a/python/paddle_serving_server_gpu/__init__.py +++ b/python/paddle_serving_server_gpu/__init__.py @@ -144,7 +144,7 @@ class Server(object): self.bin_path = os.environ["SERVING_BIN"] def check_cuda(self): - r = os.system("nvcc --version > /dev/null") + r = os.system("cat /usr/local/cuda/version.txt") if r != 0: raise SystemExit( "CUDA not found, please check your environment or use cpu version by \"pip install paddle_serving_server\"" -- GitLab