ComfyUI中解决Nvidia APEX未安装的问题

绘世启动器,一键启动,控制台,报错:
Nvidia APEX normalization not installed, using PyTorch LayerNorm。

用KIMI解决:
根据您提供的信息和搜索结果,您需要在 Windows 10 的Comfy UI中解决 Nvidia APEX 未安装的问题
以下是详细的步骤:

1. 安装 Nvidia APEX前提条件

• 确保已安装 CUDA 和 CUDNN,并且版本与 PyTorch 兼容。
• 检查电脑的 CUDA 版本和 PyTorch 内的 CUDA 版本是否一致,不一致则需要升级。

注意:安装步骤须在项目环境下的CMD中进行

2. 克隆 APEX 仓库【gitcode没有,自己镜像过去了】:

git clone https://github.com/NVIDIA/apex
git clone https://gitcode.com/shaken/apex

3. 切换到 apex 文件夹:cd apex

4. 安装 APEX:

命令:python setup.py install
验证安装
• 在 Python 中导入 apex 模块,如果没有报错,则安装成功:from apex import amp

5.重启,绘世启动器,一键启动,无报错。

ComfyUI是一个用于PyTorch的简化封装,它提供了一种更直观的界面来使用深度学习库。若要在安装ComfyUI的同时启用CUDA支持(比如Apex库,它是对PyTorch的优化工具集,特别是针对GPU加速的混合精度训练),你可以按照以下步骤操作: 1. 首先,确保你已经安装PyTorch和CUDA。如果你还没有安装CUDA,可以从NVIDIA官网下载对应版本的CUDA Toolkit。 2. 安装Python环境(推荐使用Anaconda或Miniconda)并激活虚拟环境,这有助于管理依赖项: ``` conda create -n myenv python=3.x torchvision cudatoolkit=your_version conda activate myenv ``` 3. 然后安装ComfyUI及其对CUDA的支持(例如Apex): ```bash pip install comfyui pip install apex ``` 4. 如果需要Apex的cuDNN版本适配,请确认已安装正确的cuDNN库并与CUDA版本匹配。在某些情况下,可能需要手动指定cuDNN路径: ```bash CUDA_HOME=$(python -c "import torch; print(torch.utils.cudnn.get_cuda_home())") pip install apex -f https://download.pytorch.org/whl/cu${CUDA_VERSION}/apex.zip --global-option="--cpp_ext" --global-option="--cuda_ext" --install-headers "$CUDA_HOME/include" ``` 5. 安装完成后,可以验证Apex是否成功安装以及能否在你的环境中正常工作: ```python import torch from apex import amp print(torch.__version__) print(amp.__version__) ``` 记得替换`your_version`为你实际的CUDA Toolkit版本号。如果你遇到任何问题,检查CUDA和cuDNN版本兼容性,以及权限设置可能会有所帮助。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值