这个是因为pytorch3d跟cuda的版本不一样,直接重新安装pytorch3d即可
并且需要从源码安装,很快的,国内预计5分钟就能够安装好
# 卸载当前的pytorch3d
pip uninstall pytorch3d
# 从源码直接安装
pip install "git+https://github.com/facebookresearch/pytorch3d.git"
参考资料:
pytorch3d/INSTALL.md at main · facebookresearch/pytorch3d (github.com)