概述
首先安装虚拟环境和依赖
conda create -n pytorch3d python=3.8
conda activate pytorch3d
conda install -c pytorch pytorch=1.7.1 torchvision cudatoolkit=10.2
conda install -c fvcore -c iopath -c conda-forge fvcore iopath
wget https://github.com/NVIDIA/cub/archive/1.10.0.tar.gz
tar xzf 1.10.0.tar.gz
export CUB_HOME=$PWD/cub-1.10.0
然后git clone代码并且安装
git clone https://github.com/facebookresearch/pytorch3d.git
cd pytorch3d && pip install -e .
在python里import pytorch3d即可
最后
以上就是跳跃抽屉为你收集整理的pytorch3d在linux下安装的全部内容,希望文章能够帮你解决pytorch3d在linux下安装所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复