ComfyUI+RX5700XT+Ubuntu25.04运行配置

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的仓库配置

bash 复制代码
# 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

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

bash 复制代码
# 更新包列表(默认的仓库不包含rocm,安装驱动仓库后,现在应该包含 AMD 仓库)
sudo apt update

# 现在可以通过新立得安装 ROCm
sudo apt install synaptic -y

装好后打开,右边搜索rocm,在中间的列表中找到rocm,右键选择"标记以便安装",我这里已经安装了,所以是重新安装。

然后点击上面的应用直接下载安装即可。

安装好后可以验证下:

bash 复制代码
sudo rocminfo

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

安装剩下的

简便起见,直接写个sh更容易看也方便一键执行

可以根据自己的rocm版本,在pytorch官网配置正确的下载路径,记录下来,替换步骤4中的链接

bash 复制代码
#!/bin/bash
echo "=== RX 5700 XT ComfyUI 自动配置 ==="
echo "1. 安装 Python 环境..."
sudo apt install python3 python3-venv python3-pip -y

echo "2. 克隆 ComfyUI..."
git clone https://github.com/comfyanonymous/ComfyUI.git
cd ComfyUI

echo "3. 创建虚拟环境..."
# ComfyUIEnv这是我自己起的名字,可以自行替换
python3 -m venv ComfyUIEnv
source ComfyUIEnv/bin/activate

echo "4. 安装 PyTorch..."
# 这个路径可能要自己换,我的是6.4
python3 -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.4

echo "5. 安装 剩余的ComfyUI 依赖..."
python3 -m pip install -r requirements.txt

echo "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=0

echo "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
EOF

chmod +x start_comfyui.sh

echo "配置完成!"
echo "运行 ./start_comfyui.sh 启动 ComfyUI"

启动

运行 ./start_comfyui.sh 启动 ComfyUI

相关推荐
SEO_juper7 小时前
AEO终极指南:步步为营,提升内容的AI可见性
人工智能·ai·seo·数字营销·aeo
Danceful_YJ9 小时前
31.注意力评分函数
pytorch·python·深度学习
Elastic 中国社区官方博客10 小时前
Elasticsearch:如何为 Elastic Stack 部署 E5 模型 - 下载及隔离环境
大数据·数据库·人工智能·elasticsearch·搜索引擎·ai·全文检索
逻极12 小时前
AI 规范驱动开发“三剑客”深度对比:Spec-Kit、Kiro 与 OpenSpec 实战指南
人工智能·驱动开发·ai·agent
逻极12 小时前
Claude Code 实战:Spec-Kit、Kiro、OpenSpec 规范驱动开发三剑客
ide·人工智能·驱动开发·ai·自动化
AI_567814 小时前
AI开发革命:PyCharm科学计算模式重塑TensorFlow调试体验
人工智能·ai·neo4j
文言一心16 小时前
SenseVoice 离线部署指南(Xinference Docker v1.12)
运维·docker·ai·容器
熬夜敲代码的小N17 小时前
仓颉ArrayList动态数组源码分析:从底层实现到性能优化
数据结构·python·算法·ai·性能优化
draracle17 小时前
意识、AGI与人类文明的黄昏
ai·agi·哲学