安装Pytorch3d有感

由于代码需要,需要安装pytorch3d,但真是花了很大的功夫才搞好,在此做个总结。

本来是按照官方教程去下载的,但是不知道为何,总在我已配置好的环境内重新下载cpu版本的pytorch和对应的cpu版本的pytorch3d(甚至刚开始都没发现pytorch被偷偷替换了,只发现安装了pytorch3d之后torch.cuda直接没了…)。

查看了官网的支持,发现最多支持到pytorch2.2.0,难道是因为我的pytorch版本是2.2.1?于是重新安装pytorch2.2.0,但是发现错误依然在,甚至官网的
pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu113_pyt1110/download.html替换压根没有对应源。

怀疑是目前使用的CUDA12.1适配下的pytorch2.2.0压根没有提供预先编译好的wheel,确实是安装pytorch太安逸了,只想一步到位。

最后还是选择Build from source!直接下载github上的压缩包,cd pytorch3d && pip install -e .,终于搞完!

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
05-25 1053

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值