配置SE-SSD环境

首先,最重要的是配置环境!!!

这里使用 pytorch=1.2 CUDA=10.0 CUDNN=7.5.0

CUDA下载地址:https://developer.nvidia.com/cuda-10.0-download-archive
CUDNN也在NVIDIA官网下载
然后,使用Anaconda 创建pytorch1.2环境,检查CUDA和CUDNN版本

下载所需的依赖包:

首先,确认CMake>3.13 (下载地址:https://cmake.org/download/,解压,创建软连接,

mv cmake-3.13.2-Linux-x86_64 /opt/cmake-3.13.2
ln -sf /opt/cmake-3.13.2/bin/*  /usr/bin/)

接着,下载spconv ,nuscenes-devkit和Det3D

spconv :

$ 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 *

python setup.py bdist_wheel可能会出错,检查不到CUDA,CUDNN的路径,参考博客https://blog.csdn.net/Left_control_right/article/details/109161641,如何仍然发现不了CUDNN,可以重新安装一遍CUDNN,重新运行python setup.py bdist_wheel,先删除build文件夹。
安装期间,如果缺少包,使用 “pip install 包名 ” 进行安装,有些报错,包名与错误名不一致,百度即可,如pip install scikit-build
nuscenes-devkit:

$ pip install nuscenes-devkit

Det3D:

$ git clone https://github.com/poodarchu/Det3D.git
$ cd Det3D
$ python setup.py build develop

SE-SSD:

$ git clone https://github.com/Vegeta2020/SE-SSD.git
$ cd ./SE-SSD/det3d/core/iou3d
$ python setup.py install

(pytorch版本过高,此步可能会报错,将报错文件的AT_CHECK改为TORCH_CHECK)

$ cd ./SE-SSD
$ python setup.py build develop

$ git clone https://github.com/jackd/ifp-sample.git
$ pip install -e ifp-sample
  • 1
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值