Запуск локальной AutoTrain под Anaconda
У меня собрана система с 3 различными видеокартами NVIDIA под управлением Windows 10.
Из нескольких вариантов установки AutoTrain полностью сработала непосредственная установка под Windows используя консоль Anaconda 3.
Установка под conda описана тут: Краткое руководство по локальной установки AutoTrain Advanced
Тут опишу особенности запуска AutoTrain под conda.
Запустит консоль Anaconda. Чтобы активировать эту среду, используйте
$ conda activate autotrain
Чтобы деактивировать среду, нужно подать следующую команду:
$ conda deactivate
После активации среды autotrain, надо задать переменные среды следующими командами. Например, параметры авторизации:
$ conda env config vars set HF_USERNAME=<ваше-hf-имя пользователя> $ conda env config vars set HF_TOKEN=<ваш токен-hf-write-token>
Для настройки среды GPU нужны следующие команды.
Определить текущий список видеокарт можно следующей командой.
$ nvidia-smi -L
У меня получился следующий вывод
GPU 0: NVIDIA GeForce RTX 3050 (UUID: GPU-75e8ac30-515b-177c-e07a-16430d2e3b28) GPU 1: NVIDIA GeForce GTX 1080 (UUID: GPU-26a09c99-25ec-0218-28ba-6fb8c8c15615) GPU 2: NVIDIA GeForce RTX 4060 Ti (UUID: GPU-7adf6c64-1983-fb74-dba6-e01f23f1932f)
Исходя из этого настраиваю конфигурацию драйвера cuda:
$ conda env config vars set CUDA_VISIBLE_DEVICES=2,0,1 $ conda env config vars set CUDA_DEVICE_ORDER=FASTEST_FIRST
Далее можно запускать, как в режиме веб-интерфейса, так и консольное использование. На данный момент локальный веб-интерфейс имеет глюк, который приводит к сбою авторизации, поэтому лучше запускать консольную тренировку.
$ autotrain --config CONFIG_FILE