Windows下Pytorch3d的安装方法
我的笔记本是windows GPU 安装Pytorch3d失败,废了九牛二虎之力才安装好,为了纪念我的坚持我分享给大家。
Pytorch3d在WIN下的安装
一、问题:
近期在入手三维重建中,发现很多项目程序都需要Pytorch3d库,其官网定义为:
PyTorch3D provides efficient, reusable components for 3D Computer Vision research with PyTorch.
可知Pytorch3d库可以提供多种应用于3D视觉处理的组件,其在ubuntu下安装较为简单,但win下稍繁琐,本贴是对个人安装方法的记录。
推荐内容
编辑编辑编辑
二、安装方法
安装要点:cuda、pytorch、NVIDIA CUB 版本的严格对应
安装所需链接:
- Pytorch3d 各版本:Pytorch3d Releases Download
- NVIDIA CUB 下载:NVIDIA/cub Download
- 上述PyTorch官网历史版本链接( Previous PyTorch Versions | PyTorch)里面,找到v1.11.0的安装命令, 此处选择的是cuda11.3对应的PyTorch安装命令
安装前,首先要确定版本对应关系,本人的安装版本为:
CUDA Toolkit 11.3 + pytorch 1.11.0 + Visual Studio 2019 + Pytorch3d 0.6.2 + Python 3.10
1. 对于NVIDIA CUB, 其与CUDA Toolkit的版本关系如下图:
本人安装cuda版本为11.3,所以CUB选择了对应的1.11.0版本,下载解压后将其路径添加到环境变量中:
2. 安装依赖项:
conda install -c fvcore -c iopath -c conda-forge fvcore iopath
3. 选择Pytorch3d版本,每个版本的Pytorch3d下有注明其适用的pytorch版本,如下图:
上图即pytorch3d 0.6.2版本适用的Pytorch版本为1.10.1, 1.10.2, 1.11.0,适用的Python为3.10, 但不再支持Pytorch1.6.0和Python3.6,其他版本的torch仍依据此下载对应的pytorch3d版本
4. 安装VS 2019后,打开如下图所示的"x64 Native Tools Command Prompt for VS 2019"终端,然后cd到pytorch3d解压后的目录路径里:
依次输入:
在Windows上的命令提示符(Command Prompt)中设置环境变量:
set DISTUTILS_USE_SDK=1
set PYTORCH3D_NO_NINJA=1
在Windows上的PowerShell中设置环境变量:
$env:DISTUTILS_USE_SDK = 1
在Linux和macOS上设置环境变量:
export DISTUTILS_USE_SDK=1
then, 修改 pytorch3d里setup.py文件的源码:将extra_compile_args = {“cxx”: [“-std=c++14”]} 修改为: extra_compile_args = {“cxx”: []};
5. 最后在"x64 Native Tools Command Prompt for VS 2019"终端输入:
python setup.py install --user
安装过程本人电脑上大约长达5分钟左右,页面不停的乱跳,最后的最后的最后最后显示成功:文章来源:https://www.toymoban.com/news/detail-769002.html
如果你还安装失败,请联系我文章来源地址https://www.toymoban.com/news/detail-769002.html
到了这里,关于Windows GPU 下Pytorch3d的安装方法的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!