3D目标检测算法配置——SA-SSD和spconv配置

Dependencies

Github Source

• python3.5+
• pytorch (作者用1.1.0，待会我们用1.4.0)
• opencv
• shapely
• mayavi
• spconv (v1.0, 8da6f967fb9a054d8870c3515b1b44eca2103634)

MiniConda

装过conda或者miniconda可以跳过这一步，最好装在自己目录下的local文件夹，可以参考我装CUDA的博客。

wget -c https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh
chmod +x Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh# 记住更新PATH到~/.bashrc时选no

防止conda污染原本的环境，一定一定一定要记住，不要把conda加到环境变量里，2020.12.24安装的时候只有一个地方需要输入（也就是问你要不要加到环境变量里），直接输入no就可以。如果之后我们要启动conda，我们可以用如下命令

source <path_to_your_conda_installation>/bin/activate# 默认进入base
source <path_to_your_conda_installation>/bin/activate <your_env_name># 进入指定环境
# 换源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --remove channels defaults  # optional, may fix network problem sometimes
conda config --set show_channel_urls yes
conda config --set channel_priority flexible

进入base以后我们就可以正常使用conda list等命令。

spconv 1.0

其中，最难装的就是这个spconv，其他的直接pip install一下就好了。本来我用的是本地CUDA+venv来装环境，结果有很多莫名其妙的BUG，参考MartinHahner的做法配置成功了，我用的是MiniConda，用AnaConda也是OK的（理论上，但是我尝试了下失败了），步骤如下：

下载