问题: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

好文推荐

评论可见,请评论后查看内容,谢谢!!!
 您阅读本篇文章共花了: