提交 a261e5dc authored 作者: Frederic's avatar Frederic

call nvcc only if we init this device.

上级 fc8ea431
......@@ -54,9 +54,11 @@ def execute(execute=True, verbose=True, M=2000, N=2000, K=2000,
print 'Numpy dot module:', numpy.dot.__module__
print 'Numpy location:', numpy.__file__
print 'Numpy version:', numpy.__version__
print 'nvcc version:'
subprocess.call(("nvcc", "--version"))
print
if (theano.config.device.startswith("gpu") or
theano.config.init_gpu_device.startswith("gpu")):
print 'nvcc version:'
subprocess.call(("nvcc", "--version"))
print
a = theano.shared(numpy.ones((M, N), dtype=theano.config.floatX,
order=order))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论