pytorch3d踩坑日记

1. 镜像源

一开始把清华镜像源放在的最前面,然后下载各种torch以及想要用的包都容易有依赖冲突。所以把镜像源给改成了 -defaults在前面

vim ~/.condarc
channels:
  - defaults
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - conda-forge
show_channel_urls: true

2. 先安装pytorch3d(关键)

先安装pytorch3d(这一步很关键,就是创建好虚拟环境以后直接就这一步,不要先去安装torch等其他东西),而且要指定torch为cuda版本,不然他会默认下载为cpu版本。最好先search一下,指定build来安装。

# 这里得使用pytorch3d的通道,不然找不到可下载依赖
conda search pytorch3d=0.6.0 -c pytorch3d --info

根据自己的python版本, 想安装的cuda版本来选择build

# 根据自己的环境确定build号
conda install pytorch3d=0.6.0=build号 -c pytorch3d

在确认安装以前最好看一下它要下载的torch的build是啥版本的,这样子安装的pytorch3d就是可以用的,不会报Not compile with GPU的错误

3. 其他依赖包不匹配的话

有可能有的依赖包不匹配,比如torchvision会下成cpu版本的,就卸载重新下。
下好以后,运行要跑的程序,缺啥补啥。尽量conda装,conda没有再用pip装。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
05-25 1086
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值