安装pytorch3d 遇到的问题,以及解决方案
遇到的问题
ImportError: /usr/lib/libcudart.so.10.1: version `libcudart.so.10.1' not found (required by /home/user/anaconda3/envs/pytorch_11/lib/python3.7/site-packages/pytorch3d/_C.cpython-37m-x86_64-linux-gnu.so)
ImportError: /home/user/anaconda3/envs/pytorch_10_1/lib/python3.7/site-packages/pytorch3d/_C.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZN3c104impl23ExcludeDispatchKeyGuardC1ENS_11DispatchKeyE
解决方案:
一、首先参考了https://github.com/facebookresearch/detectron2/issues/157
1、uninstall conda env and your detectron2 gcc build by the following command
$ conda remove --name your_conda_name --all
detectron2$ rm -rf build
detectron2$ rm -rf **/**.so
2、update base conda:
conda update -n base -c defaults conda
3、update conda's libgcc package:
conda update libgcc
二、Pytorch3d 官网安装教程https://github.com/facebookresearch/pytorch3d/blob/master/INSTALL.md#3-install-from-a-local-clone
1、创建conda python3.8的虚拟环境
conda create -n pytorch3d python=3.8
conda activate pytorch3d
conda install -c pytorch pytorch=1.7.0 torchvision cudatoolkit=10.2
conda install -c conda-forge -c fvcore fvcore
2、如果使用到了cuda,需要使用下面这个命令安装所需要的库
conda install -c bottler nvidiacub
3、安装pytorch3d
conda install pytorch3d -c pytorch3d