win/ubuntu pointnet2-ops安装

pointnet2-ops安装心得

第二次安装pointnet2-ops,依然遇到不少坑,特此记录!

##基础环境配置 annaconda+pytorch(>=1.4)+CUDA
一般github开源的算法多数在torch1.7.0或1.7.1,或更高版本,我这里用的pytorch1.10和1.11都可以安装,其中CUDA版本与之对应即可。

##遇到报错需要检查的几点
###1. Visual studio 的安装
尽量选择2017版本,安装包的内容如下:
在这里插入图片描述
(通用Windows平台这里不勾选是否可行没有验证)
###2. 环境变量(此电脑-属性-高级系统设置-高级-环境变量-系统变量)
(1)conda的环境变量
安装后系统变量的PATH路径里一般会有两行默认的环境变量,如图。
在这里插入图片描述

(2)CUDA环境变量
安装后一般也会有两行默认的环境变量,如图。
在这里插入图片描述
(3)VS环境变量
安装后一般也会有两行默认的环境变量INCLUDE和LIB,其中包含的系统路径大概率不全(缺少SDK的路径),需要手动添加,完整的如图。(这里主要是windows kits 10,若依然报错可尝试添加windows kits 8相关的环境变量)
在这里插入图片描述
在这里插入图片描述
(4)rc文件的拷贝
如遇报错: LINK : fatal error LNK1158: 无法运行“rc.exe”
error: command ‘C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\VC\Tools\MSVC\14.16.27023\bin\HostX86\x64\link.exe’ failed with exit code 1158。
需要将C:\Program Files (x86)\Windows Kits\10\bin\10.0.17763.0\x64中的rc.exe和rcdll.dll复制到
C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\bin
(5)gcc g++ git需要安装以及配置相应的环境变量
在保证环境配置兼容,VS安装完整,环境变量正确的情况下,安装应该就比较顺利了。

##Ubuntu22.04 下安装:
(0)确保显卡驱动,cuda,ubuntu内核及torch的版本对应;
(1)在cuda安装前安装gcc,g++;
(2)安装ninja,pyproject包
(3)如安装成功但运行报错:CUDA kernel failed: no kernel image is available for execution
错误含义为C++程序没有识别到cuda,在顺利安装cuda和torch并能进行其他gpu运算,可以确认是pointnet2编译出问题。
解决:torch环境终端输入 python -c "import torch; print(torch.cuda.get_device_capability())"检查架构算力,如输出为(8,6)则算力为8.6,并在setup文件中 os.environ[“TORCH_CUDA_ARCH_LIST”] = “5.0;6.0;6.1;6.2;7.0;7.5;” 这一行加入8.6,或者删除其他的数字,因为过低的算力部分gpu并不支持。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值