未验证 提交 d4282ea9 编写于 作者: Z Zhou Wei 提交者: GitHub

fix multi cuda environment bug (#31694)

上级 09482dde
......@@ -442,7 +442,8 @@ def find_cuda_home():
[which_cmd, 'nvcc'], stderr=devnull)
if six.PY3:
nvcc_path = nvcc_path.decode()
nvcc_path = nvcc_path.rstrip('\r\n')
# Multi CUDA, select the first
nvcc_path = nvcc_path.split('\r\n')[0]
# for example: /usr/local/cuda/bin/nvcc
cuda_home = os.path.dirname(os.path.dirname(nvcc_path))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册