如何在其他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 && python setup.py install

已在RTX3090成功配置环境:python3.8+pytorch1.9.1+cuda11.1+cudnn8.0.5
在这里插入图片描述

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值