如何在其他torch和cuda框架下安装pytorch3d(运行报错:undefined symbol: _ZNK2at10TensorBase8data_ptrIdEEPT_v)

问题:pytorch3d需要指定的pytorch和cuda版本才能正常安装,不如就会报各种错误。
但是修改系统的cuda版本和torch版本将会对其他代码造成影响,需要繁琐地切换。
如何在非官方指定的gpu环境安装pytorch3d?

比如:

ImportError:
/home/xxx/anaconda3/envs/xxx/lib/python3.8/site-packages/pytorch3d/_C.cpython-38-x86_64-linux-gnu.so:
undefined symbol: _ZNK2at10TensorBase8data_ptrIdEEPT_v

解决方法:

不要使用官方文档推荐直接安装

conda install pytorch3d -c pytorch3d
conda install pytorch3d -c pytorch3d-nightly
pip install pytorch3d

下载原始代码,自行安装

git clone https://github.com/facebookresearch/pytorch3d.git
conda activate your_env
cd pytorch3d && pip setup.py install

已在RTX3090成功配置环境:python3.8+pytorch1.9.1+cuda11.1+cudnn8.0.5
如何在其他torch和cuda框架下安装pytorch3d(运行报错:undefined symbol: _ZNK2at10TensorBase8data_ptrIdEEPT_v)_第1张图片

你可能感兴趣的:(pytorch,python,深度学习)