我的环境Ubuntu18 Cuda11.1 pytorch1.91 python3.8 显卡Nvidia3080Ti
默认安装
pip install pytorch3d
默认安装的是pytorch0.3.0,调用时报错
ImportError: libcudart.so.10.1: cannot open shared object file: No such file or directory
解决办法
去pytorch3d的github上面下载满足版本需求的pytorch3d,我这里是0.7.2
https://github.com/facebookresearch/pytorch3d/releases
我下载的Source code,然后解压到本地,再
python setup.py install