【linux】安装pytorch3d

写在开头

不要看网上的博客安装,直接参考官方安装文档去安装。

坑1:安装pytorch3d后torch.cuda用不了

使用下面的命令安装后

conda install pytorch3d

会提示安装下面的包,注意pytorch和torchvision都是cpu版本的,这样就会覆盖我们原本cuda版本的pytorch和torchvision,导致之后发现torch.cuda.is_available()返回False
在这里插入图片描述

解决办法1:所有都用conda安装

开始安装torch时就用conda安装,不要用pip安装
之后安装pytorch3d也用conda安装

解决办法2:从源码安装pytorch3d

看似最复杂,其实最有效
第一步:git clone
第二步:编译安装

git clone https://github.com/facebookresearch/pytorch3d.git
cd pytorch3d && pip install -e .

猜你喜欢

转载自blog.csdn.net/weixin_43693967/article/details/132155655