Этот пример поможет научиться запускать задачи обучения на GPU.
Модель обучается с помощью библиотеки PyTorchна датасете MNIST путем создания и отправки задачи обучения.
Для запуска примера:
-
Создайте новый Jupyter Server или подключитесь к уже существующему Jupyter Server, используя документацию.
-
Создайте новую папку и загрузите следующие файлы через веб-интерфейс Jupyter Server на платформе ML Space:
- mnist.npz— датасет с рукописными цифрами;
- quick-start.ipynb — Jupyter-ноутбук для загрузки на сервер;
- requirements.txt — файл с зависимостями, который используется для сборки кастомного контейнера;
- tensorflow_mnist_estimator.py — код модели на
Pytorch.
-
Запустите ноутбук quick-start.ipynb в интерфейсе Jupyter Server.