写在前面
算法和代码框架均来自港大的课题组
gen6d介绍页面
我只是拿来做一个本科毕设,答辩结束后会遵循GPL协议开放所有代码
这篇博客仅作为开发日志,记录一下部署过程中的各种问题
环境问题
- readme中的推荐python版本为3.8,但是当前较新的CUDA版本要求3.9+,要么安装旧版CUDA,要么用python3.12.9和新版CUDA
- 命令nvidia-smi查看的是显卡信息,而不是CUDA信息,使用指令nvcc --version才能确认当前安装的Nvidia tool kit版本
- requirements.txt中要安装
pytorch
,但是实际上pytorch的包名是torch
,安装pytorch
会报错 - 由于图方便,试图在Windows环境下部署该项目,但是发现
pytorch3d
并不支持Windows,需要自己做源码编译,这实在没有必要,明天换到Linux环境再试试