Установите более высокую общую память, чтобы избежать RuntimeError с PyTorch в Google Colab

Question

Использование pytorch 1.0 Preview с fastai v1.0 в Colab.

Я часто получаю RuntimeError: DataLoader worker (pid 13) is killed by signal: Bus error. для более ресурсоемких задач (ничего огромного).

Похоже, проблема с общей памятью: https://github.com/pytorch/pytorch/issues/5040#issue-294274594

Исправление выглядит так, как будто оно изменяет общую память в контейнере Docker:

https://github.com/pytorch/pytorch/issues/2244#issuecomment-318864552

Похоже, что общая память док-контейнера не была установлена ​​достаточно высоко. Задание большего значения путем добавления --shm-size 8G к команде запуска docker, кажется, уловка, как упомянуто здесь.

Как я могу увеличить общую память док-контейнера, работающего в Colab, или иным образом избежать этой ошибки?

Answer 1

Невозможно изменить этот параметр в colab, но значение по умолчанию было поднято, чтобы решить эту проблему, поэтому вам не нужно менять этот параметр далее: https://github.com/googlecolab/colabtools/issues/329