安装适配于服务器的pytorch、pytorch3d_查看pytorch3d版本

记录一下安装版本合适的pytorch、pytorch3d的方法

首先新建一个conda环境，进入新环境

输入nvidia-smi查看服务器的版本号：

nvidia-smi

在pytorch官网查看适配于服务器的pytorch版本： 

Start Locally | PyTorch

以我的服务器为例，cuda版本是11.5，pytorch给了cuda为11.3，11.6的选项，考虑到版本向下兼容，采用11.3的选项：

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

接着安装torchvision

conda install torchvision

安装pytorch3d，按照官网的指南pytorch3d/INSTALL.md at main · facebookresearch/pytorch3d · GitHub

conda install -c fvcore -c iopath -c conda-forge fvcore iopath
conda install -c bottler nvidiacub
conda install jupyter
conda install pytorch3d -c pytorch3d

大功告成！