目录
使用 second.pytorch做3D目标检测的可视化,需要安装spconv,然后就开始了漫长的踩坑路。。。这里记录下来,方便自己以后查阅。
spconv1.2安装步骤
补充:后面我又用pytorch1.3.1安装spconv1.2版本,原因是不想使用系统路径下的cuda而转用个人用户下的cuda(如何装可以参考我的博客)。详细步骤参考下面1.0版本
步骤一:首先需要在CMakeList.txt中添加以下cuda路径,添加在第五行,添加完如下图所示。(或者在setup.py的cmake_args中添加'-DCMAKE_CUDA_COMPILER=/home/gye/cuda100/bin/nvcc')
set(CMAKE_CUDA_COMPILER "/home/gye/cuda100/bin/nvcc")
步骤二:将下面路径文件中usr相关的依赖项全部改成自己路径下的,然后重新运行应该就可以了。如果报错error: ‘RegisterOperators’ is not a member of ‘torch’,将对应文件所在行的中的torch::RegisterOperators改为torch::jit::RegisterOperators即可。
vim ~/anaconda3/envs/PCdet/lib/python3.6/site-packages/torch/share/cmake/Caffe2/Caffe2Targets.cmake