写在开头
不要看网上的博客安装,直接参考官方安装文档去安装。
坑
坑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 .