dust3r部署踩坑全记录

目前dust3r是三维重建最新最好的技术,运用了ViT编码器、Transformer、注意力机制、回归等技术,无需相机参数标定。

但是我部署过程中有很多坑,记录一下。

1.OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.

我之前装好了cuda和cudnn,最开始是完全按照readme.md里面的方式部署,用了conda,新建一个环境,环境装了Python3.11。但是安装到

复制代码
python setup.py build_ext --inplace

这一步时就会报错。我查了一下,说是Anaconda安装的cudatoolkit不完整,需要下载完整的Nvidia CUDA Toolkit。似乎要在虚拟环境再装一个cudatoolkit。但是网上找的解决方案并不能解决问题。

后来我考虑到虚拟环境装不了,试试不用Anaconda,于是在另一台电脑上,没有装Anaconda,直接装的Python3.8,然后直接装,这一步居然没报错,过去了。Python3.8后面会有报错,但是有办法解决。而我又装了Python3.12和Python3.11,还是会报这个错。还有就是Python3.8没有能装起来pytorch和pytorch-cuda,报错好像是说没有找到满足条件的库,但是不影响后面运行。

2.ImportError: cannot import name 'cache' from 'functools' (D:\Python38\lib\functo
ols.py)

这就是前面说的Python3.8的报错,查了一下Python3.8及以前的版本没有这个,用Python3.9就没问题。但是我装了Python3.12和Python3.11又有了第一个报错。Python3.8及以前的版本用的不是cache,而是叫lru_cache的,名字变了一下而已。于是我就用Python3.8,把报错的dust3r\dust3r\cloud_opt\init_im_poses.py里面的cache 全部改成lru_cache(共两处),然后就可以run了。

3.TypeError: 'int' object is not subscriptable

网上查了一下,说是int型不可以被引用。因为报错的地方是assert(dust3r\dust3r\model.py),于是我直接把这句话注释掉了,就可以run了。

4.File "demo.py", line 294, in <module>
model = AsymmetricCroCo3DStereo.from_pretrained(weights_path).to(args.device)

因为之前按readme下载了DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth在新建的checkpoints文件夹里面,而demo.py读取的地方是

复制代码
weights_path = "naver/" + args.model_name
复制代码
args.model_name就是DUSt3R_ViTLarge_BaseDecoder_512_dpt。
故weights_path应为checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth。
解决方案:将这一行改为
复制代码
weights_path = "checkpoints/" + args.model_name + ".pth"
复制代码
然后就可以run了。

展示一下效果:

相关推荐
_不会dp不改名_10 分钟前
leetcode_1382 将二叉搜索树变平衡树
算法·leetcode·职场和发展
文火冰糖的硅基工坊31 分钟前
《投资-93》价值投资者的认知升级与交易规则重构 - 衡量公司价值的本质是在公司的整个存续期间能够创造多少自由现金流,而不是当下有多少现金流。
重构·架构·产业链
lpfasd12332 分钟前
从OpenAI发布会看AI未来:中国就业市场的重构与突围
人工智能·重构
greentea_201338 分钟前
Codeforces Round 173 B. Digits(2043)
c++·算法
xiaohanbao091 小时前
Transformer架构与NLP词表示演进
python·深度学习·神经网络
m0_743106462 小时前
LOBE-GS:分块&致密化效率提升
人工智能·算法·计算机视觉·3d·几何学
徐子童2 小时前
优选算法---字符串
java·算法·字符串·笔试·高精度相乘
西瓜啵啵奶茶2 小时前
LeetCode 热题 100 : 普通数组
算法
jikiecui2 小时前
信奥崔老师:C++ 程序设计入门
算法
Q741_1472 小时前
C++ 位运算 高频面试考点 力扣 面试题 17.19. 消失的两个数字 题解 每日一题
c++·算法·leetcode·面试·位运算