在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. 官方给出的安装指南一定要仔细看,并要注意相关的版本信息
    在Ubuntu20.04系统anaconda3的环境下安装pytorch3d+opencv_第1张图片
  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,实际运行命令为
    在Ubuntu20.04系统anaconda3的环境下安装pytorch3d+opencv_第2张图片
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并不需要安装 ↩︎

你可能感兴趣的:(学习教程/笔记,pytorch,opencv,3d)