Найти в Дзене
24 подписчика

Выбор компьютера для обучения нейросетей


Стоит ли использовать ноутбук и из каких комплектующих собирать десктоп?

⛔️ Не повторяйте чужих ошибок!

💻 Macbook Apple M1

Большую часть времени я работаю дома, но мне крайне необходима возможность в любой момент сложить своё рабочее место в рюкзак. Поэтому много лет мой макбук висит под рабочим столом, подключен к огромному дисплею, беспроводной клавиатуре и мыши.

Сейчас это Macbook Pro 14 на процессоре M1pro с 16Gb памяти. Все проекты начинаю на нем, потом, при необходимости подключаю внешние серверы с мощными GPU.

👍 При обучении нейронок макбук не перегревается, хотя все ядра утилизируются почти на 100%.

О выборе не жалею, но сейчас взял бы модель с 32Gb оперативной памяти, посколько на этом чипе она же является и памятью GPU. Также стоит выбирать максимальную версию процессора с максимальным количеством ядер.

Но, если с бюджетом проблемы, то можно обойтись и самым дешевым Macbook Air на M1. Поскольку для полноценной работы все-равно можно и нужно использовать дополнительные средства, например Google Collab или свой сервер с GPU.

🖥 Стационарный компьютер/сервер

При сборке десктопа, если вы решили, что он вам необходим, я бы сейчас исходил из следующих принципов:

1️⃣ Требования TensorFlow

– Центральный процессор должен поддерживать AVX. Обычно Core i3/i5/i7/i9 поддерживают, а процессоры Pentium и Celeron до Tiger Lake – нет.
– Карта NVIDIA с архитектурой CUDA® 3.5, 5.0, 6.0, 7.0, 7.5, 8.0 и выше.

2️⃣ Высокая скорость обучения

Подумайте, какой смысл собственной сборки, если компьютер будет работать на уровне бесплатного Colab? По этой причине я бы рассматривал десктопы с картами RTX3080 и более мощными.

3️⃣ Память GPU не менее 12Gb

Учтите, что даже простой transfer learning с картинками 300x300px в датасете может запросто откусить 12Gb и более. Вы будете сильно ограничены в экспериментах, если приобретете GPU с меньшим объемом ОЗУ...


Выбор компьютера для обучения нейросетей  Стоит ли использовать ноутбук и из каких комплектующих собирать десктоп?  ⛔️ Не повторяйте чужих ошибок!
1 минута
334 читали