点云检测 SECOND 网络环境配置
1.如果有docker镜像可以用,建议别自己配置了,可能有点耗时
2.所成功配置并可运行的框架如下:
- 最好在Anaconda下建立个虚拟环境
- cmake 3.15.2
- python3.7
- cuda 10.1
- cudnn7.6.3
- pytorch 1.2
- pytorchvision 0.2.2
- sparse conv
- 依赖包有(中间有import error 可再用pip或者conda安装)
- scikit-image
- numba
- fire
- opencv-python
3.环境变量设置
export PATH="/home/yifei/anaconda3/bin:$PATH"
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
export PATH="/home/yifei/anaconda3/bin:$PATH"
export PATH=$PATH:/usr/bin/cmake
export PYTHONPATH=$PYTHONPATH:/home/yifei/code/second.pytorch/
export NUMBAPRO_CUDA_DRIVER=/usr/lib/x86_64-linux-gnu/libcuda.so
export NUMBAPRO_NVVM=/usr/local/cuda/nvvm/lib64/libnvvm.so
export NUMBAPRO_LIBDEVICE=/usr/local/cuda/nvvm/libdevice
4.配置网络参数
根据文章以及所给代码修改路径以及相应配置文件即可!
大概问题有这么多,有问题欢迎多交流!