问题: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 && python setup.py install
已在RTX3090成功配置环境:python3.8+pytorch1.9.1+cuda11.1+cudnn8.0.5