概述
记录一下安装版本合适的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大功告成！
最后
以上就是洁净哈密瓜为你收集整理的安装适配于服务器的pytorch、pytorch3d的全部内容，希望文章能够帮你解决安装适配于服务器的pytorch、pytorch3d所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复