Запуск TensorFlow тесты в Google Colab

Вопрос задан: 9 месяцев назад Последняя активность: 9 месяцев назад
up 1 down

Я хочу, чтобы запустить тесты на Google Colab для обеспечения воспроизводимости, но я получаю ошибку системы в конце, что я не на моей локальной машине.

Я настроил TensorFlow в Google Colab с

!pip install tensorflow==1.12.0
import tensorflow as tf
print(tf.__version__)

который после некоторых линий установки, гравюр:

1.12.0

Затем я хочу, чтобы запустить простой тест:

import tensorflow as tf

class Tests(tf.test.TestCase):

    def test_gpu(self):
        self.assertEqual(False, tf.test.is_gpu_available())

tf.test.main()

Тест проходит (вместе с тестовой сессией по умолчанию) на моей локальной машине, а также на Colab, но после того, что ядро ​​возвращает ошибку системы:

..
----------------------------------------------------------------------
Ran 2 tests in 0.005s

OK
An exception has occurred, use %tb to see the full traceback.

SystemExit: False

После вызова %tb, Я получаю длинные трассировки стеки вставленные ниже, что дает мало информации. Как я могу это исправить?

StackTrace является:

SystemExit                                Traceback (most recent call last)
<ipython-input-20-6a87bf6320f2> in <module>()
      7         self.assertEqual(False, tf.test.is_gpu_available())
      8 
----> 9 tf.test.main()
     10 
     11 

/usr/local/lib/python3.6/dist-packages/tensorflow/python/platform/test.py in main(argv)
     62   """Runs all unit tests."""
     63   _test_util.InstallStackTraceHandler()
---> 64   return _googletest.main(argv)
     65 
     66 

/usr/local/lib/python3.6/dist-packages/tensorflow/python/platform/googletest.py in main(argv)
     98       args = sys.argv
     99     return app.run(main=g_main, argv=args)
--> 100   benchmark.benchmarks_main(true_main=main_wrapper)
    101 
    102 

/usr/local/lib/python3.6/dist-packages/tensorflow/python/platform/benchmark.py in benchmarks_main(true_main, argv)
    342     app.run(lambda _: _run_benchmarks(regex), argv=argv)
    343   else:
--> 344     true_main()

/usr/local/lib/python3.6/dist-packages/tensorflow/python/platform/googletest.py in main_wrapper()
     97     if args is None:
     98       args = sys.argv
---> 99     return app.run(main=g_main, argv=args)
    100   benchmark.benchmarks_main(true_main=main_wrapper)
    101 

/usr/local/lib/python3.6/dist-packages/tensorflow/python/platform/app.py in run(main, argv)
    123   # Call the main function, passing through any arguments
    124   # to the final program.
--> 125   _sys.exit(main(argv))
    126 

/usr/local/lib/python3.6/dist-packages/tensorflow/python/platform/googletest.py in g_main(argv)
     68   if ('TEST_TOTAL_SHARDS' not in os.environ or
     69       'TEST_SHARD_INDEX' not in os.environ):
---> 70     return unittest_main(argv=argv)
     71 
     72   total_shards = int(os.environ['TEST_TOTAL_SHARDS'])

/usr/lib/python3.6/unittest/main.py in __init__(self, module, defaultTest, argv, testRunner, testLoader, exit, verbosity, failfast, catchbreak, buffer, warnings, tb_locals)
     93         self.progName = os.path.basename(argv[0])
     94         self.parseArgs(argv)
---> 95         self.runTests()
     96 
     97     def usageExit(self, msg=None):

/usr/lib/python3.6/unittest/main.py in runTests(self)
    256         self.result = testRunner.run(self.test)
    257         if self.exit:
--> 258             sys.exit(not self.result.wasSuccessful())
    259 
    260 main = TestProgram

SystemExit: False

1 ответ

Возможно, для Вашего проекта будут необходимы бесплатные векторные карты. На нашем сайте представлены карты для всех стран.

Реклама

up 1 down accepted

Ошибка вы видите приходит от UnitTest пытается выйти из процесса питона, который Jupyter предотвращает от вашего имени. Вы можете избежать этого с т.д .:

import tensorflow as tf
class Tests(tf.test.TestCase):
    def test_gpu(self):
        self.assertEqual(False, tf.test.is_gpu_available())
import unittest
unittest.main(argv=['first-arg-is-ignored'], exit=False)

(Обратите внимание, что последняя строка отличается от твоего, и поднимается из https://github.com/jupyter/notebook/issues/2746)