提交 9d63d0ed authored 作者: notoraptor's avatar notoraptor

Fix GPU initialization checking on buildbot.

上级 55e94968
...@@ -54,7 +54,9 @@ mkdir %BUILDBOT_DIR% ...@@ -54,7 +54,9 @@ mkdir %BUILDBOT_DIR%
echo "Directory of stdout/stderr %BUILDBOT_DIR%" echo "Directory of stdout/stderr %BUILDBOT_DIR%"
REM Exit if theano.gpuarray import fails REM Exit if theano.gpuarray import fails
python -c "import theano.gpuarray; theano.gpuarray.use('%DEVICE%')" || exit /b REM NB: As we define `set THEANO_FLAGS=init_gpu_device=cuda` above,
REM importing theano.gpuarray will automatically initialize GPU.
python -c "import theano.gpuarray; exit(1 - int(theano.gpuarray.pygpu_activated))" || exit /b
REM Fast run and float32 REM Fast run and float32
set FILE=%BUILDBOT_DIR%\theano_python2_fastrun_f32_tests.xml set FILE=%BUILDBOT_DIR%\theano_python2_fastrun_f32_tests.xml
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论