Pytorch3D安装全流程-亲自安装

简单介绍

Pytorch3D是一个用于加速深度学习在处理3D相关信息时候的运算速率的库。深度学习有时会需要处理大量的3D数据,比如在人脸建模的时候。这时候使用这些库会大大帮助我们节省运算成本。

遇到问题

按照官方的安装方法会遇到不少问题:https://github.com/facebookresearch/pytorch3d/blob/main/INSTALL.md。我个人在安装的时候,输入相关的指令并不能够成功安装,也是折腾了快一个上午加一个中午才解决,这里分享一下,希望能帮助更多的人。

准备工作

1.conda创建好的虚拟环境

首先你需要一个用conda创建好的虚拟环境,如何创建网上教程很多,在此不多赘述。注意你的cuda里面要有cudatoolkit这个包,这个包是用来使用cuda的,简单来说就是调用你电脑GPU的。

要记好你虚拟环境的python/pytorch版本和cudatoolkit版本,建议pytorch版本手动设定安装版本,否则会自动安装到最新版,最新版往往与别的很多包都是不兼容的

2.cub安装配置

cub的版本一定要和cudatoolkit版本对应,版本对应说明的链接在这里:https://gitcode.net/mirrors/nvidia/cub?utm_source=csdn_github_accelerator

知道自己版本是什么了之后,开始下载,下载链接:https://github.com/NVIDIA/cub/releases

请添加图片描述

进入网页后往下翻,找到对应版本的Assets。这里有两个压缩包,一般如果是windows电脑的话选择zip,如果是linux的话选择tar.gz(然后用命令解压)

之后解压相应安装包。

解压完成后,将文件所在位置目录添加到环境变量中

环境变量设置方式:右键我的电脑(此电脑)—属性—高级系统设置—环境变量—Path变量——添加新环境变量

3.MinGW安装

网上有博客说要安装,因为我电脑里之前需要用vscode所以安装过,电脑里就有,所以我也不确定是否一定要装。

正式安装

1.首先下载pytorch3D安装包

下载链接:https://github.com/facebookresearch/pytorch3d/releases

一定要注意版本对应!!

请添加图片描述

比如上面这张图,就说明pytorch3D 0.7.1 版本只支持1.9.0到1.12.1的pytorch

下载后进行解压,建议安装到site-packages这个文件夹里,因为这个文件夹是专门存放pytorch各种包的地方,比如我的:.\anaconda3\Lib\site-packages

2.更改相关的文件

解压好的文件并不能直接使用,需要先改一下代码。

打开pytorch3D文件夹(我的文件名是pytorch3d-0.7.2),里面有个setup.py文件夹,修改下面部分的代码:

请添加图片描述

52行我注释掉的是原来的代码,53行有红点的是修改后的。我猜这里修改是不指定C++的编译器。因为其实python的底层代码还是C++,这个包下载好并不能直接用,需要安装编译后这个包才能用。

网上有的教程改了很多地方的代码,但是我自己只修改了这一个地方就可以使用了。如果只改了一个没有成功,参考这篇博客(点击链接)

3.安装VS2019

需要安装vs2019,这里上网找下教程即可,教程很多。下载vs2019的原因是用来编译下载的安装包。

等下载安装好之后,找到这玩意儿x64 Native Tools Command Prompt for VS 2019(如果是win10win11可以直接在搜索栏里搜索)。这个是vs2019自带的终端。

4.安装!!!

终于到这一步了。

首先依次在终端窗口输入下面的命令:

set DISTUTILS_USE_SDK=1
set PYTORCH3D_NO_NINJA=1

然后再输入这个对pytorch3d进行编译安装:

python setup.py install

我记得当时终端里跑了好多好多行代码,应该是在一行行编译,大概过了个十五分钟,就成功了。

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
青葱年少的头像青葱年少普通用户
上一篇 2023年10月19日
下一篇 2023年10月19日

相关推荐