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

make check_blas.py print the nvcc version.

上级 5d568c17
...@@ -17,6 +17,7 @@ import os ...@@ -17,6 +17,7 @@ import os
import sys import sys
import time import time
from optparse import OptionParser from optparse import OptionParser
import subprocess
import numpy import numpy
import theano import theano
...@@ -53,6 +54,8 @@ def execute(execute=True, verbose=True, M=2000, N=2000, K=2000, ...@@ -53,6 +54,8 @@ def execute(execute=True, verbose=True, M=2000, N=2000, K=2000,
print 'Numpy dot module:', numpy.dot.__module__ print 'Numpy dot module:', numpy.dot.__module__
print 'Numpy location:', numpy.__file__ print 'Numpy location:', numpy.__file__
print 'Numpy version:', numpy.__version__ print 'Numpy version:', numpy.__version__
print 'nvcc version:'
subprocess.call(("nvcc", "--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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论