EC2のUbuntu上にComifyUIの環境構築をする

※ 一通り記載したのでアップしますが後ほどアップデートします

本記事の目的

AWSのEC2上にComifyUI環境を作成したので軌跡をメモ

インスタンス情報
マシンイメージ: Ubuntu Server 22.04
インスタンスタイプ: g5.4xlarge # 生成AIを動かすときはVRAMが大きいほうが良いので選定複数GPUがゆうこうなら12xlargeに上げても良さそう
ストレージ: 128GB # モデルがでかいので多めにストレージを取っておく

ComifyUIのリポジトリ:
https://github.com/comfyanonymous/ComfyUI

環境構築

まずは挨拶のaptアップデート

sudo apt update

Python関係の導入

sudo apt install python3.10-venv

sudo apt install python3-pip

NvidiaのGPUを使えるようにする

使えるドライバーを確認

sudo apt install -y ubuntu-drivers-common
ubuntu-drivers devices

上記コマンドでrecommendedされたバージョンに以下のバージョン 535のところ を変更して実行

sudo apt install -y nvidia-driver-535

ComifyUI

リポジトリをクローンしてそのディレクトリに移動

git clone https://github.com/comfyanonymous/ComfyUI.git
cd ComfyUI

リポジトリの指示にしたがって環境構築
まずはpytorchのインストール

pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121

requirements.txt のパッケージをインストール

pip install -r requirements.txt

これで環境構築完了！！

実行

main.py を実行してサーバーの起動

python3 main.py