ubuntu安装pytorch3d

2021/5/10 7:29:10

本文主要是介绍ubuntu安装pytorch3d,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

第一步

# Install latest from GitHub
pip install -U 'git+https://github.com/facebookresearch/fvcore'
pip install -U 'git+https://github.com/facebookresearch/iopath'
https://github.com/NVIDIA/cub/archive/refs/tags/1.10.0.tar.gz

第二步

pip install scikit-image matplotlib imageio plotly opencv-python
pip install black 'isort<5' flake8 flake8-bugbear flake8-comprehensions

第三步

注意对应Python版本和cuda版本,pytorch版本

pip install pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py37_cu101_pyt171/download.html


这篇关于ubuntu安装pytorch3d的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程