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

相关推荐
小雨青年4 小时前
MateChat 进阶实战:打造零后端、隐私安全的“端侧记忆”智能体
前端·华为·ai·华为云·状态模式
Keep_Trying_Go5 小时前
基于Zero-Shot的目标计数算法详解(Open-world Text-specified Object Counting)
人工智能·pytorch·python·算法·多模态·目标统计
paopao_wu6 小时前
腾讯HunyuanOCR开源端到端OCR-1B:本地部署与测试
ai·开源·ocr·hunyuan
西柚小萌新7 小时前
【深入浅出PyTorch】--9.使用ONNX进行部署并推理
人工智能·pytorch·python
LDG_AGI7 小时前
【推荐系统】深度学习训练框架(十):PyTorch Dataset—PyTorch数据基石
人工智能·pytorch·分布式·python·深度学习·机器学习
带刺的坐椅8 小时前
Solon AI 开发学习10 - chat - 工具调用概念介绍
ai·llm·solon·tool-call
on_pluto_9 小时前
【debug】解决 5070ti 与 pytorch 版本不兼容的问题
人工智能·pytorch·python
心态与习惯10 小时前
使用 pytorch 进行深度学习的一般步骤
pytorch·深度学习·流程·步骤
蓝耘智算11 小时前
如何选择合适的GPU算力服务商?企业AI算力采购指南
ai·gpu算力·蓝耘
蓝耘智算11 小时前
蓝耘元生代GPU算力调度云平台深度解析:高性价比算力云与GPU算力租赁首选方案
人工智能·ai·gpu算力·蓝耘