提交 5acd763e authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix bug in displaying error message.

上级 98f09cbd
......@@ -127,7 +127,7 @@ if cuda_available:
def use(device, force=False):
global cuda_enabled
global cuda_enabled, cuda_initialization_error_message
if force and not cuda_available and device.startswith('gpu'):
raise EnvironmentError("You forced use of device %s, but CUDA initialization failed "
"with error:\n%s" % (device, cuda_initialization_error_message))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论