提交 d1aa9482 authored 作者: David Warde-Farley's avatar David Warde-Farley

Explicit check for GPU availability once CUDA is available.

上级 d2249db8
......@@ -91,6 +91,8 @@ except Exception, e:
if cuda_available:
cuda_available=device_available()
if not cuda_available:
warning('CUDA is installed, but GPU device is not available')
if cuda_available:
#check if their is an old cuda_ndarray that was loading instead of the one we compiled!
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论