提交 5de2bd22 authored 作者: Frederic's avatar Frederic

use the nvcc path found by theano.

上级 a261e5dc
...@@ -57,7 +57,8 @@ def execute(execute=True, verbose=True, M=2000, N=2000, K=2000, ...@@ -57,7 +57,8 @@ def execute(execute=True, verbose=True, M=2000, N=2000, K=2000,
if (theano.config.device.startswith("gpu") or if (theano.config.device.startswith("gpu") or
theano.config.init_gpu_device.startswith("gpu")): theano.config.init_gpu_device.startswith("gpu")):
print 'nvcc version:' print 'nvcc version:'
subprocess.call(("nvcc", "--version")) subprocess.call((theano.sandbox.cuda.nvcc_compiler.nvcc_path,
"--version"))
print print
a = theano.shared(numpy.ones((M, N), dtype=theano.config.floatX, a = theano.shared(numpy.ones((M, N), dtype=theano.config.floatX,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论