DenseFusion复现-可以在30系/40系显卡运行

库行僧

已于 2024-04-01 13:07:35 修改

阅读量1k

点赞数 1

文章标签：深度学习 pytorch

于 2023-09-10 15:49:02 首次发布

本文链接：https://blog.youkuaiyun.com/m0_53397118/article/details/132791555

版权

作者电脑显卡为4060，因为使用DenseFusion作者pytorch1.0的代码没有成功，发现很多人在30系显卡上复现失败，经过查资料后发现是因为cuda版本与显卡算力不匹配，需要提高cuda版本，因此也需要提高pytorch版本，后来在GitHub找到了pytorch1.1.7.1版本的代码，之后复现成功。

代码链接：

kuaitoukid/DenseFusion: "DenseFusion: 6D Object Pose Estimation by Iterative Dense Fusion" code repository (github.com)

环境搭建：

笔者conda环境下所用到的包：

这边贴出一部分：

python=3.8.8 cudatoolkit=11.0.221 cudnn=8.0.5 pytorch=1.7.1  torchvision=0.8.2=py38_cu110  pyyaml=6.0 scipy=1.10.1  pillow=9.3.0 opencv-python=4.8.0.76 numpy=1.24.3

在conda环境下,DenseFusion复现时需要用到的包资源-优快云文库https://download.youkuaiyun.com/download/m0_53397118/88321743直接利用conda安装即可，我看到有人说30系显卡最低的cuda版本需要11.1，但是我在cuda=11.0下运行成功了。