本文旨在复现SE-SSD,由于本人非计算机专业出身,复现过程中,走了很多弯路,现在将这些问题一一列举出来以及需要参考的一些关键资料;切记切记千万不要使用非python3.6,非官方推荐的版本,真的是坑一个比一个大!!!!!!!(本文市作者回忆写出,具体有所疏忽还望大家见谅)
本人复现的系统是Ubuntu20.04
关于环境:
建议使用SE-SSD所提供的环境,尽量不要使用其他环境,经过本人实操,感觉上Ubuntu的系统环境版本差距不大。
1.第一步,安装CudaToolK cuDNN【Linux】在一台机器上同时安装多个版本的CUDA(切换CUDA版本)_linux安装多个cuda-CSDN博客直接参考该博客即可,需要注意的是系统Ubuntu的环境选择最新即可
作者使用的是Cuda 10.0以及cuDNN7.6.5
2.安装gcc 版本的
如果gcc的版本大于7的话,那么系统会报错 no later than 7
安装gcc 对于新手推荐使用 自动安装的方式
ubuntu系统查看gcc版本及版本切换_ubuntu查看gcc版本-CSDN博客
3.安装spconv1.0 安装卷积模块
$ sudo apt-get install libboost-all-dev
$ git clone https://github.com/poodarchu/spconv --recursive
$ cd spconv && python setup.py bdist_wheel
$ cd ./dist && pip install *
给大家看看一个成功的图
4. cmake的安装版本请使用 3.14.4
5. opencv-python 和opencv-contrib-python 版本号清保持一致,
4.TORCH_CHECK’ was not declared in this scope 该错误直接将错误问文件改成AT_CHECK 即可
5. Not found: 'python/cv2/py.typed' 这个错误直接升级pip
pip install --upgrade pip
6.setuptools.installer is deprecated. Requirements should be satisfied by a PEP 517
如果报出该项错误,有两种情况,一种是setuptool工具需要升级;第二种是手动安装所需的包即可;
例如手动安装,skbuild 直接 可以 添加镜像源 建议国内用清华
pip install scikit-build
7.RegisterOperators’ is not a member of ‘torch’ 直接安装提示更改即可
torch 后面添加一个jit即可
8.
给大家一个可以参考的包的版本,以txt给出,作为附件供大家下载
其他资料:
CUDA、cuDNN以及pytorch的版本选择和下载_cuda官网选择版本-CSDN博客Python版本与opencv版本的对应关系_opencv-python对应版本-CSDN博客