nVidia Tesla P40使用anaconda本地重编译pytorch3d成功加载ComfyUI-3D-Pack

背景

自己用的是nVidia Tesla P40,垃圾佬专属卡

使用下面的由YanWenKun提供的ComfyUI-3D-Pack预安装环境,但在本地编译pytorch3d这一步出错,后面有出错信息,如果有和我一样的卡一样的问题,参看此文的解决方法

老版本的ComfyUI-3D-Pack预集成包
https://github.com/YanWenKun/ComfyUI-Windows-Portable
2024/12/14后切换到新仓库
https://github.com/YanWenKun/Comfy3D-WinPortable

在这里插入图片描述
我使用的预集成包的版本是2025/1/21
https://github.com/YanWenKun/Comfy3D-WinPortable/releases
在这里插入图片描述
解压后目录结构如下:
在这里插入图片描述
其中的python_standalone目录就是预安装完的库,包括独立的python环境
对于首次需要本地编译的TORCH_CUDA_ARCH_LIST的参数,
在这里插入图片描述
我的Tesla P40查显卡根据架构编号查资料需要设置为6.1,和bat文件内默认的设置相同

set TORCH_CUDA_ARCH_LIST=6.1

架构编号查看链接如下:
https://arnon.dk/matching-sm-architectures-arch-and-gencode-for-various-nvidia-cards/

在这里插入图片描述

在我的3070上按下面B站UP主啦啦啦的小黄瓜详细的步骤是可以成功的,但是P40就一直在编译pytorch3d的部分失败
3070的根据架构编译参数是8.6
无奈只能自己用anaconda重新构建一个python环境,这样可以方便的用于不同python环境版本的环境切换

编译pytorch3d时的错误信息如下:
在这里插入图片描述
出错信息
在这里插入图片描述

在这里插入图片描述
最终编译出错,导致comyui-3d-pack import failed,其他几个必须本地编译的库都可以正常

ComfyUI-3D-Pack节点加载失败
在这里插入图片描述
在这里插入图片描述

解决思路

步骤1、获取预安装包的python_standalone里的python环境,确定所需要的每一个python库的版本
方法是在解压后的预安装环境中使用 python -m pip list > requirement.txt 导出只有版本号
在这里插入图片描述
使用 python -m pip freeze > requirement.txt 导出,还带有预安装环境构建时的安装方式
在这里插入图片描述

步骤2、使用anaconda构建一个完全相同的python环境
将所有的python库分成如下几种安装方式,按顺序在conda环境中进行安装:

  1. 直接pip install 库==版本号 方式安装的
  2. 需要添加cuda版本进行选择安装方式 安装命令最后需要加上相应的torch cuda链接的
  3. 使用预构建包 whl文件安装的
  4. 必须本机编译安装的(构造包里extra目录下的除 pytorch3d外的5个)

几个关键的版本号

  1. Python – 3.12.8
  2. Pytorch – 2.5.1
  3. Cuda – 12.4
  4. xformers – 0.0.29.post1
    在这里插入图片描述

本文基础

需要在操作系统中已经成功安装完成如下环境,这部分内容不在本文讨论范围

  1. cuda 12.4

  2. cudnn 复制库函数到上面的cuda目录

  3. VS build环境 包括VC++库 WinSDK

  4. anaconda环境 相当于自动包括了所有的python版本

系统环境需要注意的几点

  1. 显卡驱动cuda版本可以高于用的cuda库的版本

  2. 如果需要安装多个cuda库的版本,从最低版

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值