提交 41e9bc5e 编写于 作者: B bjjwwang

rename 1027 to 102

上级 39f89a0c
...@@ -73,7 +73,7 @@ if (SERVER) ...@@ -73,7 +73,7 @@ if (SERVER)
set(VERSION_SUFFIX 101) set(VERSION_SUFFIX 101)
elseif(CUDA_VERSION EQUAL 10.2) elseif(CUDA_VERSION EQUAL 10.2)
if(CUDNN_MAJOR_VERSION EQUAL 7) if(CUDNN_MAJOR_VERSION EQUAL 7)
set(VERSION_SUFFIX 1027) set(VERSION_SUFFIX 102)
elseif(CUDNN_MAJOR_VERSION EQUAL 8) elseif(CUDNN_MAJOR_VERSION EQUAL 8)
set(VERSION_SUFFIX 1028) set(VERSION_SUFFIX 1028)
endif() endif()
......
...@@ -429,7 +429,7 @@ class Server(object): ...@@ -429,7 +429,7 @@ class Server(object):
if device_type == "0": if device_type == "0":
device_version = self.get_device_version() device_version = self.get_device_version()
elif device_type == "1": elif device_type == "1":
if version_suffix == "101" or version_suffix == "1027" or version_suffix == "1028" or version_suffix == "112": if version_suffix == "101" or version_suffix == "102" or version_suffix == "1028" or version_suffix == "112":
device_version = "gpu-" + version_suffix device_version = "gpu-" + version_suffix
else: else:
device_version = "gpu-cuda" + version_suffix device_version = "gpu-cuda" + version_suffix
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册