提交 46b95d95 authored 作者: Gabe Schwartz's avatar Gabe Schwartz

Fixed python3 compat issue introduced in 80ebe9c3.

Bytes need to be decoded prior to performing any string operations, such as split/strip, on them.
上级 bf9156bc
...@@ -99,7 +99,7 @@ def is_nvcc_available(): ...@@ -99,7 +99,7 @@ def is_nvcc_available():
"""Return True iff the nvcc compiler is found.""" """Return True iff the nvcc compiler is found."""
def set_version(): def set_version():
p_out = output_subprocess_Popen([nvcc_path, '--version']) p_out = output_subprocess_Popen([nvcc_path, '--version'])
ver_line = decode(p_out[0].split('\n')[-1]) ver_line = decode(p_out[0]).strip().split('\n')[-1]
build, version = ver_line.split(',')[1].strip().split() build, version = ver_line.split(',')[1].strip().split()
assert build == 'release' assert build == 'release'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论