Linux Mint зависает при интенсивном использовании видеокарты NVIDIA?

Question

[ideo]

Всем привет.
Я недавно начал заниматься изучением нейронных сетей. Использую библиотеку pytorch на python3 в среде jupyter notebook. Соответственно моей задачей является обучить нейронную сеть, используя 3 слоя cnn и 2 слоя fc. Таким образом на видеокарте выделяется 3.5GB. Я использую видеокарту своего ноутбука GeForce 950M 4GB.
Проблемы начинают тогда, когда я отчищаю память с одного notebook. Linux Mint 18.2 напрочь зависает и его никак не вывести, кроме как тупым выключением, нажатием кнопки выкл (ctrl+shift+f1 тоже не помогает, он просто не откликается на эту команду).
Можно ли исправить данное поведение ноутбука? Является ли это проблемой железа?

update: драйвер на Линукс теперь не видит карточку.
Проверил карточку на Винде с помощью FurMark. Все окей. Ноут не перезагрузился, а карточка жива. Также проверил на лиунксе этой программой тоже все окей.

Answer 1

[Dimonchik]

хз насчет железа,но поведение видухи ооочень сильно зависит от драйверов

а драйвера видеокарт под линух делают задней ногой - у геймеров винда

Answer 2

[xdgadd]

Тут может быть дело в pytorch'e. Пару дней назад один чел в telegram чате AI Community жаловался на точно такую же проблему. Утверждает, что баг в самой библиотеке.

Comments

[ideo]

Тоже читал там, что чувак жаловался, что у него память утекает. Как я понял у него проблемы были с архитектурой.

[xdgadd]

ideo, ну, он написал "кажется, я нашёл баг". Баг может быть как в архитектуре, так и в либе.

[ideo]

xdgadd, на самм деле ты подал хорошую идею. Попробую описать проблему на форуме pytorch'a. У них хорошее комьюнити, как я понял, когда искал для себя информацию по гиперпараметрам.