以Ubuntu搭配Nvidia顯示卡為例。
1. 安裝CUDA#
安裝 Nvidia專有驅動
安裝 CUDA
確認Nvidia驅動版本
nvidia-smi
- 確認CUDA版本
nvcc --version
2. 安裝依賴套件#
- 安裝Python、Git
sudo apt install python3 python3-pip python3-virtualenv wget git
- 安裝 Anaconda,此套件的用意在於鎖定Python版本。
3. 安裝ComfyUI#
- 用conda建立Python 3.10.6的環境
conda create --name comfyui python=3.10.6
- 啟動虛擬環境:
conda activate comfyui
- 複製ComfyUI儲存庫
cd ~
git clone https://github.com/comfyanonymous/ComfyUI.git
cd ComfyUI
- 安裝PyTorch CUDA
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu124
- 安裝Python依賴套件
pip install -r requirements.txt
將safetensors存檔點模型放到
~/ComfyUI/models
目錄。啟動程式,之後用瀏覽器開啟
http://127.0.0.1:8188
python main.py
註解:如果你之後想用Bash指令稿啟動ComfyUI,可以將以下conda的鉤子加入指令稿頂部,這樣就可以自動進入conda環境。
eval "$(conda shell.bash hook)"
conda activate comfyui
4. 如何更新程式#
拉取最新變更
cd ~/ComfyUI
git pull