ComfyUI+RX5700XT+Ubuntu25.04运行配置

前言

有台机器，是AMD RX5700XT的显卡，近期想要试试能否跑ComfyUI，在Windows 11上尝试了CPU环境运行的，出一张图需要10小时以上，简直不要太慢，本着简便的想法尝试在WSL2上启用ROCm，发现WSL2微软并没有为其内核启用ROCm支持，一直报错，不太想重新编译内核，于是安装了Ubuntu25.04，准备直接在原生系统跑起来，后来发现这一套配置有不少坑，配置起来比较麻烦，于是记录下，给有同样烦恼的朋友提供一份思路。

大体步骤

1. 安装使用 Ubuntu 25.04
2. 在官网下载安装AMD的驱动仓库配置
3. 通过新立得安装 ROCm
4. ROCm和HIP环境变量配置
5. 使用 PyTorch 官网的torch安装命令 - 确保版本匹配自己的系统
6. 安装剩余的ComfyUI依赖

详细步骤

安装驱动，添加ROCm的仓库配置

# 1. 访问 AMD 官方 Linux 驱动页面
# https://www.amd.com/en/support/download/linux-drivers.html# 2. 选择显卡型号（Radeon RX 5000 Series -> RX 5700 XT）
# 3. 下载对应的驱动安装包（通常是一个小的 .deb 文件）
# 4. 安装这个驱动包，这个包只是一个仓库配置，所以很小
sudo dpkg -i amdgpu-install_6.4.60402-1_all.deb

安装ROCm

这里用新立得安装，不然有很多细碎的依赖不完整，导致容易安装不成功

# 更新包列表（默认的仓库不包含rocm，安装驱动仓库后，现在应该包含 AMD 仓库）
sudo apt update# 现在可以通过新立得安装 ROCm
sudo apt install synaptic -y

装好后打开，右边搜索rocm，在中间的列表中找到rocm，右键选择“标记以便安装”，我这里已经安装了，所以是重新安装。
然后点击上面的应用直接下载安装即可。

安装好后可以验证下：

sudo rocminfo

如果出现Agent2，里面有显卡型号，那就是OK了。

安装剩下的

简便起见，直接写个sh更容易看也方便一键执行
可以根据自己的rocm版本，在pytorch官网配置正确的下载路径，记录下来，替换步骤4中的链接

#!/bin/bash
echo "=== RX 5700 XT ComfyUI 自动配置 ==="
echo "1. 安装 Python 环境..."
sudo apt install python3 python3-venv python3-pip -yecho "2. 克隆 ComfyUI..."
git clone https://github.com/comfyanonymous/ComfyUI.git
cd ComfyUIecho "3. 创建虚拟环境..."
# ComfyUIEnv这是我自己起的名字，可以自行替换
python3 -m venv ComfyUIEnv
source ComfyUIEnv/bin/activateecho "4. 安装 PyTorch..."
# 这个路径可能要自己换，我的是6.4
python3 -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.4echo "5. 安装 剩余的ComfyUI 依赖..."
python3 -m pip install -r requirements.txtecho "6. 创建启动脚本..."
cat > start_comfyui.sh << 'EOF'
#!/bin/bash
export ROCM_PATH=/opt/rocm
export PATH=$ROCM_PATH/bin:$PATH
export LD_LIBRARY_PATH=$ROCM_PATH/lib:$ROCM_PATH/lib64:$LD_LIBRARY_PATH# RX 5700 XT 特定配置
# ！！！！这里的配置是针对我的机器，其他的型号可以去AMD驱动支持下载页面查
export PYTORCH_ROCM_ARCH="gfx1010"
export HSA_OVERRIDE_GFX_VERSION=10.3.0
export ROC_ENABLE_PRE_VEGA=1# HIP 配置
export HIP_PLATFORM=amd
export HIP_VISIBLE_DEVICES=0echo "ROCm 环境已配置"
echo "PYTORCH_ROCM_ARCH: $PYTORCH_ROCM_ARCH"
echo "HSA_OVERRIDE_GFX_VERSION: $HSA_OVERRIDE_GFX_VERSION"source ComfyUIEnv/bin/activate
python main.py --gpu-only --listen 0.0.0.0 --port 8188
EOFchmod +x start_comfyui.shecho "配置完成！"
echo "运行 ./start_comfyui.sh 启动 ComfyUI"

启动

运行 ./start_comfyui.sh 启动 ComfyUI