记录一下安装版本合适的pytorch、pytorch3d的方法
首先新建一个conda环境,进入新环境
输入nvidia-smi查看服务器的版本号:
复制代码
1nvidia-smi
在pytorch官网查看适配于服务器的pytorch版本:
Start Locally | PyTorch
以我的服务器为例,cuda版本是11.5,pytorch给了cuda为11.3,11.6的选项,考虑到版本向下兼容,采用11.3的选项:
复制代码
1conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
接着安装torchvision
复制代码
1conda install torchvision
安装pytorch3d,按照官网的指南pytorch3d/INSTALL.md at main · facebookresearch/pytorch3d · GitHub
复制代码
1
2
3
4conda install -c fvcore -c iopath -c conda-forge fvcore iopath conda install -c bottler nvidiacub conda install jupyter conda install pytorch3d -c pytorch3d
大功告成!
最后
以上就是洁净哈密瓜最近收集整理的关于安装适配于服务器的pytorch、pytorch3d的全部内容,更多相关安装适配于服务器内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复