提交 296299b6 编写于 作者: M MRXLT

fix cuda check

上级 b67e7c8b
......@@ -211,8 +211,13 @@ class Server(object):
self.bin_path = os.environ["SERVING_BIN"]
def check_cuda(self):
r = os.system("cat /usr/local/cuda/version.txt")
if r != 0:
cuda_flag = False
r = os.popen("whereis libcudart.so")
r = r.read().split(":")[1]
if "cudart" in r and os.system(
"ls /dev/ | grep nvidia > /dev/null") == 0:
cuda_flag = True
if not cuda_flag:
raise SystemExit(
"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.
先完成此消息的编辑!
想要评论请 注册