Win10下pytorch3D安装方法,本人亲测可用

news/2024/11/24 3:10:45/

本人的电脑配置如下:

GPU:GTX 1050TiCUDA:11.1python:3.8pytorch:1.9.0pytorch3d:0.7.1NVIDIA CUB:cub-1.9.10-1

目录

1.版本确认

2.下载cub

3.下载pytorch3d

 4.下载VSC2019

 5.编译pytorch3d源文件


1.版本确认

        这个步骤非常非常重要,因为不同的pytorch3d版本适配于不同的torch版本,你可以进到facebook/pytorch3d的官网查看不同的适配版本(这是0.7.1的),其余的你们可以自己查看https://github.com/facebookresearch/pytorch3d/releases/tag/v0.7.1

 此外,你要保证你的CUDA环境变量要对,比如我的CUDA是11.1,我的cuda toolkit也是11.1的,并把它添加到你的用户的环境变量里,此外还有你的conda也要添加到用户环境变量里面,因为后面我们会用到一个叫做Microsoft Visual Studio2019的工具,在使用时需要你的conda命令进行激活。如果你们有和我一样的版本,cuda11.1,你可以自己新建一个环境,然后重新装一个torch:

pip --default-timeout=1000 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

其它依赖:

conda install -c fvcore -c iopath -c conda-forge fvcore iopath

这样我们就有了一个1.9.0的torch。

 你们也可以自己输入这段代码验证:

import torch
print(torch.cuda.is_available())
print(torch.__version__)

2.下载cub

        CUDA和NVIDIA CUB版本对应关系如下:GitHub - NVIDIA/cub: Cooperative primitives for CUDA C++.

下载好后,将其加入自己的环境变量:

3.下载pytorch3d

        根据你自己的需要下载对应版本的pytorch3d压缩包

下载好后,解压放在文件夹,我是放在了e盘 

 4.下载VSC2019

        因为下载好的源文件需要用c++进行编译VS2019是最好的编译的版本,我们可以去官网下载,安装时,我们选择默认的C++的安装选项

 5.编译pytorch3d源文件

        这个部分是最最关键的,我个人是这么觉得的!

第一步:找到x64 tool vs2019 prompt:

 注:千万要用管理员的方式运行,不然编译的时候头文件调用会出问题!!

 第二步:

cd 你源文件文件夹的位置

 此外,你需要进入conda环境,就像我前面有个(pytorch3d)一样!!

这就提到了我们前面说的为什么要把conda添加到环境变量里面,直接:

conda activate (你创建的环境,我这里是pytorch3d)

接着输入这两段(为了防止编译超时):

set DISTUTILS_USE_SDK=1
set PYTORCH3D_NO_NINJA=1

最后运行:

python setup.py install

x64 native tool vs2019 prompt会开始编译,差不多10分钟就可以编译好:

 到这里,pytorch3d的安装就结束啦,有其他问题的小伙伴可以评论区问我哦!


http://www.ppmy.cn/news/441108.html

相关文章

数字图像处理期末考点整理(全)

计算:傅里叶变换,双线性插值,直方图均衡化,灰度共生矩阵,霍夫曼编码,区域增长/合并,中值滤波 简答:窗口/模板处理,BMP文件存储格式,滤波器和平滑算子的特点&…

【大学计算机技术】第七章 测试3

文章目录 选择题 选择题 在局域网中,每台计算机都必须设置一个IP地址,以便进行信息帧的传送。 A. 正确 B. 错误 正确答案: B 每个网卡都有一个全球唯一的地址称为MAC地址,因此网卡比较复杂,一般做成独立插卡的形式。 …

教大家一个免费复制粘贴百度文库文字的方法

教大家一个免费复制粘贴百度文库文字的方法 chrome浏览器装个翻译插件就行了,把要复制的文字框选出来,点一下翻译,然后设置里面调中文-》中文,在复制就ok了

2023年03月六级真题全3套【可复制可划线查词】共11页PDF

2023年03月六级真题全3套【可复制可划线查词】共11页PDF 2023年03月六级真题全3套【可复制可划线查词】共11页PDF 2023年03月六级真题全3套【可复制可划线查词】共11页PDF

如何批量复制word文件名到excel?

如何批量复制word文件名到excel?批量复制word文件的名称到excel表格中,这个操作最突出的重点是批量,也就是我们常说的批量处理,批量处理能帮助我们提高效率,很多人不知道如何批量复制,那么就只能采用一个一…

批量替换 Word 文档某几页

概要:一个 Word 文档通常会包含很多的页,比如我们的论文、与客户签订的合同等这些 Word 文档往往都是几百页的。有时候在后期检查的时候,我们如果发现这些文件的某几页有问题需要批量的替换成新的页面,那我们就麻烦了!…

word复制到别的计算机,在计算机之间复制移动Word自动更正项

在计算机之间复制移动Word自动更正项 如果您的计算机中的内置自动更正项目列表中没有包含所需要的自动检测和更正的更正项目(例如打印错误、拼写错误和不正确的大写等),您可以向列表中添加项目。如果您想使用在另一台安装有MicrosoftOffice2003或MicrosoftWord2003的…

Office 中Word十大技巧

1、快速定位到上次编辑的位置 打开Word文件后,按下shiftF5键你就会发现光标已经快速定位到你上一次编辑的位置了。小提示:其实shiftF5的作用是定位到Word最后三次编辑的位置 2、快速插入当前日期或时间 有时写完一篇文章,觉得有必要在文章…