MLflowのセットアップ

以下の手順に従って、MLflowをセットアップします。

5 Minute Tracking Server Overview | MLflow
https://mlflow.org/docs/latest/getting-started/tracking-server-overview/index.html

以下コマンドの通り、ユーザ・ディレクトリを用意します。

sudo useradd mlflow -d /var/lib/mlflow
sudo mkdir /var/lib/mlflow
sudo chown mlflow:mlflow /var/lib/mlflow

Pythonの仮想環境を作成し、MLflowをインストールします。

sudo su - mlflow
cd ~
python3 -m venv venv
. venv/bin/activate
pip install mlflow

インストールできたかを、確認しておきます。

mlflow --version

systemdでのサービス化

管理しやすくするためsystemdでサービス化します。
次の設定ファイルを作ります。

/etc/systemd/system/mlflow.service

[Unit]
Description=MLflow
After=syslog.target
After=network.target

[Service]
RestartSec=2s
Type=simple
User=mlflow
Group=mlflow
WorkingDirectory=~
ExecStart=/var/lib/mlflow/venv/bin/mlflow server --host 0.0.0.0 --port 5000 --backend-store-uri /var/lib/mlflow/.mlflow
Restart=always
Environment=VIRTUAL_ENV=/var/lib/mlflow/venv
Environment=PATH=/var/lib/mlflow/venv/bin

[Install]
WantedBy=multi-user.target

設定をリロード、サービスを開始します。

sudo systemctl daemon-reload
sudo systemctl enable mlflow
sudo systemctl start mlflow

サービスを実行しているホストの5000番ポートにブラウザでアクセスすると、
MLflowの画面が表示されます。

以上。