文章目录
前言
Ubuntu16.04系统下基于pytorch的代码环境需要配置pointnet2_ops_lib。笔者虚拟环境配置为Ubuntu16.04、pytorch1.8.2、cuda11.1、python3.6。pointnet2_ops_lib源码来自erikwijmans/Pointnet2_PyTorch,配置要求在作者书写的README里说得很清楚了,仔细阅读。总体来说,虚拟环境里缺啥装啥,记录下遇到的问题及解决方法。
一、No module named ‘pointnet2_ops’
刚开始跑代码的时候,报错:No module named ‘pointnet2_ops’,搜索以后才知道这个计算子需要单独配置。在上述链接处下载代码压缩包Pointment2_PyTorch-master.zip,解压后放入相关代码文件夹里。
进入创建好的虚拟环境(笔者:py36_pt182),cd到pointnet2_ops_lib文件夹所在目录(笔者:/EdgeDetect/Pointment2_PyTorch-master),运行命令:
pip install -r requirements.txt
缺啥装啥。