在Ubuntu20.04系统anaconda3的环境下安装pytorch3d+opencv

原来的pytorch3d和opencv的编程环境莫名崩掉了,装了很久都没重新装成功,这次记录下来,以后再有问题就翻翻笔记。

Preliminaries

由于后续要用到cuda,所以首先查看当前的cuda runtime api版本1

nvcc -V

nvcc-V
这里,可以看出,当前电脑的cuda版本是11.3
【这里可能会有可能cuda版本不是11.3,后面会再写一份笔记,记录一下,如何更改cuda的版本】
在正式开始安装之前,需要先安装Anaconda3。
Linux安装Anaconda3的教程可见这里
注:安装完Anaconda3后,如不想关闭终端再打开,可以采用如下命令使安装生效

source ~/.bashrc

开始安装

这里先安装pytorch3d,后安装opencv

Pytorch3d安装

  1. 官方给出的安装指南一定要仔细看,并要注意相关的版本信息
    pytorch3d installation
  2. 创建一个python版本为3.9编程环境,并起名为pytorch3d(环境名称可以随意起),并激活该环境
conda create -n pytorch3d python=3.9
conda activate pytorch3d
  1. 参考Pytorch官方安装指南,运行建议的命令行。此时安装的Pytorch版本为1.10.1,不在Pytorch3d需求的版本中,因此在安装的时候可指定Pytorhc版本为1.10.0,实际运行命令为
    Pytorch installation
conda install pytorch=1.10.0 torchvision torchaudio cudatoolkit=11.3 -c pytorch
  1. 安装fvcore和iopath
conda install -c fvcore -c iopath -c conda-forge fvcore iopath
  1. 依赖包安装好之后2,开始安装Pytorch3d,这里选择从Anaconda云安装(仅Linux)
conda install pytorch3d -c pytorch3d

至此,Pytorch3d已经安装成功,采用conda list命令查看
pytorch3d success

Opencv安装

Opencv给出的安装指南似乎很麻烦,但在Anaconda环境中只需一行代码即可

conda install opencv

这里有可能采用以上命令安装的版本与python版本不兼容,可能的原因是数据源的问题,可以采用以下命令进行安装

pip install opencv-python

【完】


  1. cuda有runtime api和driver api,两者有对应的版本,在这里用于pytorch(3d)的是runtime api。而采用nvidia-smi查看到的是driver api,一般情况下,driver api版本>=runtime api,具体可见参考链接1参考链接2 ↩︎

  2. 由于这里是从Anaconda云安装Pytorch3d,而非源码,因此CUB library并不需要安装 ↩︎

  • 0
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值