提交 2058a9eb authored 作者: abergeron's avatar abergeron

Merge pull request #1938 from nouiz/find_cuda

Fix the detection of where cuda is installed.
...@@ -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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论