提交 6d780a78 编写于 作者: M MRXLT

fix cuda check

上级 7c4dbc0e
...@@ -144,7 +144,7 @@ class Server(object): ...@@ -144,7 +144,7 @@ class Server(object):
self.bin_path = os.environ["SERVING_BIN"] self.bin_path = os.environ["SERVING_BIN"]
def check_cuda(self): def check_cuda(self):
r = os.system("whereis cuda > /dev/null") r = os.system("nvcc --version > /dev/null")
if r != 0: if r != 0:
raise SystemExit( raise SystemExit(
"CUDA not found, please check your environment or use cpu version by \"pip install paddle_serving_server\"" "CUDA not found, please check your environment or use cpu version by \"pip install paddle_serving_server\""
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册