由于代码需要,需要安装pytorch3d,但真是花了很大的功夫才搞好,在此做个总结。
本来是按照官方教程去下载的,但是不知道为何,总在我已配置好的环境内重新下载cpu版本的pytorch和对应的cpu版本的pytorch3d(甚至刚开始都没发现pytorch被偷偷替换了,只发现安装了pytorch3d之后torch.cuda直接没了…)。
查看了官网的支持,发现最多支持到pytorch2.2.0,难道是因为我的pytorch版本是2.2.1?于是重新安装pytorch2.2.0,但是发现错误依然在,甚至官网的
pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu113_pyt1110/download.html
替换压根没有对应源。
怀疑是目前使用的CUDA12.1适配下的pytorch2.2.0压根没有提供预先编译好的wheel,确实是安装pytorch太安逸了,只想一步到位。
最后还是选择Build from source!直接下载github上的压缩包,cd pytorch3d && pip install -e .
,终于搞完!