当前位置: 首页 > news >正文

win下配置pytorch3d

一、配置好的环境:py 3.9 + pytorch 1.8.0 + cuda 11.1_cudnn 8_0 + pytorch3d 0.6.0 + CUB 1.11.0

你可能觉得pytorch3d 0.6.0版本有点低,但是折腾不如先配上用了,以后有需要再说。
(后话:py 3.9 + pytorch 1.12.1 + cuda 11.3_cudnn 8_0 + pytorch3d 0.7.1 + CUB 1.11.0 也OK的)

1.1 创建新环境

conda create -n torch3Dtorch180 python=3.9
conda activate torch3Dtorch180

1.2 找之前的pytorch版本,我这里直接给出我的选择的版本

# CUDA 11.1
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

一行行复制粘贴运行下面的命令

conda install -c fvcore -c iopath -c conda-forge fvcore iopath
conda install jupyter
pip install scikit-image matplotlib imageio plotly opencv-python
pip install black usort flake8 flake8-bugbear flake8-comprehensions

1.3 下载对应的pytroch3d,Version 0.6.0

在这里插入图片描述
解压后,在pytorch3d中源码修改以下:

1.3.1 setup.py 中43行:

extra_compile_args = {"cxx": ["-std=c++14"]},改为extra_compile_args = {"cxx": []}

1.3.2 setup.py 中60行:

-std=c++14那行注释掉

1.3.3 pytorch3d/csrc/iou_box3d.cu 18行:

#include "utils/pytorch3d_cutils.h" 那行注释掉

1.3.4 pytorch3d/csrc下,iou_utils.cuh和 iou_utils.h 中搜索not

像下面的这种语句中,把not改为!
在这里插入图片描述

1.3.5 在你环境中的torch\utils\cpp_extension.py中

本人的是E:\01_softwares\07_Anaconda3\install\envs\torch3Dtorch180\Lib\site-packages\torch\utils\cpp_extension.py

match = re.search(r'(\d+)\.(\d+)\.(\d+)', compiler_info.decode().strip())中加入 "gbk"
在这里插入图片描述

1.4 下载对应的 CUB

具体的CUB与cuda toolkit对应关系见 https://github.com/NVIDIA/cub,这里我下载的是CUB 1.11.0
在这里插入图片描述

在这里插入图片描述
同样的解压到你想要的位置,添加设置环境变量,变量名CUB_HOME,变量值即为刚才解压的cub的文件路径
在这里插入图片描述


二、安装CUDA ToolKit 11.3

这部分参考:https://blog.csdn.net/weixin_43469047/article/details/130170756?spm=1001.2014.3001.5501


三、下载VS2019

如果你有VS2022,大可以先试试,不行再下VS2019。


四、最后的配置

搜索x64即可找到x64 Native Tools Command Prompt for VS 2019终端,切换路径和虚拟环境。
在这里插入图片描述
依次执行下面命令:

set DISTUTILS_USE_SDK=1
set PYTORCH3D_NO_NINJA=1

最后执行安装pytorch3D的代码

python setup.py install

等待10分钟,就装好了
在这里插入图片描述


以上参考自:
《pytorch3D Windows下安装经验总结》
《win10系统下安装PyTorch3d》


总结一些问题:

1、

在这里插入图片描述
可能的解决方案:使用VS2019而不是VS2022,VS2019的具体小版本倒无所谓,因为我去找community社区版也只有最后一版可供下载了。
参考自:https://github.com/facebookresearch/pytorch3d/issues/1024


http://www.lryc.cn/news/58950.html

相关文章:

  • JS字符串对象
  • Linux系统对文件及目录的权限管理(chmod、chown)
  • 半透明反向代理 (基于策略路由)
  • 课前测5-超级密码
  • QML控件--Menu
  • 002:Mapbox GL更改大气、空间及星星状态
  • 2022年第十三届蓝桥杯题解(全)C/C++
  • 【cmake学习】find_package 详解
  • WEB攻防-通用漏洞PHP反序列化POP链构造魔术方法原生类
  • Baumer工业相机堡盟工业相机如何通过BGAPISDK里的图像处理库进行图像转换(C++)
  • JD开放平台接口(获得JD商品详情, 按关键字搜索商品,按图搜索京东商品(拍立淘), 获得店铺的所有商品,获取推荐商品列表, 获取购买到的商品订单列表)
  • 上海亚商投顾:沪指震荡反弹 游戏、传媒概念股再度大涨
  • C/C++ 玩转StoneValley库:从入门到精通
  • CentOS7-部署Tomcat并运行Jpress
  • 菜鸟程序员的3年心酸逆袭之旅!今天你对我爱搭不理,明天我让你高攀不起!
  • 【Scala】异常 隐式转换 泛型
  • 1673_MIT 6.828 Homework xv6 lazy page allocation要求翻译
  • 六、Locust之TaskSets详解
  • flask_知识点3_css
  • Redis_概述_特性_IO模型
  • [论文速览] Sparks of Artificial General Intelligence: Early experiments with GPT-4
  • 舔狗日记:学姐生日快到了,使用Python把她的照片做成视频当礼物
  • 从《移动互联网应用程序(App)收集使用个人信息自评估指南》看个人信息保护着力点
  • 电脑0x0000001A蓝屏错误怎么U盘重装系统教学
  • Day939.如何小步安全地升级数据库框架 -系统重构实战
  • 2023 年十大 API 管理趋势
  • 计算机网络微课堂1-3节
  • [Eigen中文文档] Array类与元素操作
  • python学习,全球有哪些特别好的社区推荐呢?
  • LC-1042. 不邻接植花(四色问题(染色法))