提交 0b32dffa authored 作者: Frederic's avatar Frederic

check_blas.py now print the theano flag device and add GPU timming.

上级 433e73d7
...@@ -43,6 +43,7 @@ def execute(execute=True, verbose=True, M=2000, N=2000, K=2000, ...@@ -43,6 +43,7 @@ def execute(execute=True, verbose=True, M=2000, N=2000, K=2000,
print ' blas.ldflags=', theano.config.blas.ldflags print ' blas.ldflags=', theano.config.blas.ldflags
print ' compiledir=', theano.config.compiledir print ' compiledir=', theano.config.compiledir
print ' floatX=', theano.config.floatX print ' floatX=', theano.config.floatX
print ' device=', theano.config.device
print 'Some environment variables:' print 'Some environment variables:'
print ' MKL_NUM_THREADS=', os.getenv('MKL_NUM_THREADS') print ' MKL_NUM_THREADS=', os.getenv('MKL_NUM_THREADS')
print ' OMP_NUM_THREADS=', os.getenv('OMP_NUM_THREADS') print ' OMP_NUM_THREADS=', os.getenv('OMP_NUM_THREADS')
...@@ -202,11 +203,12 @@ if __name__ == "__main__": ...@@ -202,11 +203,12 @@ if __name__ == "__main__":
C2075 0.25s C2075 0.25s
C1060 0.46s C1060 0.46s
GTX680 0.154s 0.218s GTX Titan(D15U-50)0.06s don't work
GTX580 0.164s 0.203s GTX680 0.12s 0.154s 0.218s
GTX480 0.192s 0.237s 0.27s GTX580 0.16s 0.164s 0.203s
GTX470 0.238s 0.297s 0.34s GTX480 0.19s 0.192s 0.237s 0.27s
GTX660 0.24s GTX470 0.23s 0.238s 0.297s 0.34s
GTX660 0.20s 0.23s
GTX560 0.30s GTX560 0.30s
GTX460 0.37s 0.45s GTX460 0.37s 0.45s
GTX285 0.452s 0.452s 0.40s # cuda 3.0 seam faster? driver version? GTX285 0.452s 0.452s 0.40s # cuda 3.0 seam faster? driver version?
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论