提交 1b859832 authored 作者: Frederic's avatar Frederic

Fix the detection of where cuda is installed.

上级 bb572061
...@@ -129,8 +129,7 @@ def find_cuda_root(): ...@@ -129,8 +129,7 @@ def find_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")):
config.cuda.root = os.path.split(dir)[0] return os.path.split(dir)[0]
return
rpath_defaults = [] rpath_defaults = []
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论