提交 21b2f50a authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Make cuda.root work correctly in the presence of paths that end with /.

上级 7b735c43
...@@ -155,7 +155,7 @@ def default_cuda_root(): ...@@ -155,7 +155,7 @@ def default_cuda_root():
return '' return ''
for dir in s.split(os.path.pathsep): for dir in s.split(os.path.pathsep):
if os.path.exists(os.path.join(dir, "nvcc")): if os.path.exists(os.path.join(dir, "nvcc")):
return os.path.split(dir)[0] return os.path.dirname(os.path.abspath(dir))
return '' return ''
AddConfigVar( AddConfigVar(
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论