安装openpcdet

本机配置

RTX3070
Ubuntu 20.4
Nvidia driver: 470
cuda: 11.1
cudnn: 8.0.5
pytorch: 1.9.1+cu111

安装cuda,对因版本的cudnn

cuda

  1. Nvidia官网下载适合自己机器的CUDA版本,如cuda_11.1.1_455.32.00_linux.run
    cuda driver, cuda, pytorch对应关系,但对应关系不是很新一定要去官网看gpu型号对应的算力,下载对应的cuda版本
    下载cuda*.run文件
    下载cuda*.run文件
  2. 安装
sudo sh cuda_11.1.1_455.32.00_linux.run

如果已经安装driver,可以不要选driver
accept
cuda toolkit自带driver,如果之前已经安装过了driver可以不用安装。(如何取消?光标位于Driver处按空格键取消选择该选项)
去除勾选driver

  1. 配置
sudo gedit ~/.bashrc
export CUDA_HOME=/usr/local/cuda
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
source ~/.bashrc
  1. 测试是否安装成功
验证
cd ~/NVIDIA_CUDA-11.1_Samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
# or
nvcc -V

sudo make 后的结果
./deviceQuery的结果

如果安装错了删除cuda

cd /usr/local/cuda-11.0/bin/
sudo ./cuda-uninstaller
sudo rm -rf /usr/local/cuda-11.0

安装cudnn

  1. 下载cudnn文件
  2. 解压
tar -xzvf cudnn-11.1-linux-x64-v8.0.5.39.tgz
  1. 复制一些文档到usr/cuda中去
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/include/cudnn_version.h /usr/local/cuda/include #一定要加不然后面安装spconv会识别不出cudnn版本
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
  1. 安装Deb包, cuDNN Runtime Library for Ubuntu20.04(Deb),cuDNN Developer Library for Ubuntu20.04(Deb),cuDNN Code Samples and User Guide for Ubuntu20.04(Deb)
sudo dpkg -i libcudnn8_8.0.5.39-1+cuda11.1_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.5.39-1+cuda11.1_amd64.deb
sudo dpkg -i libcudnn8-samples_8.0.5.39-1+cuda11.1_amd64.deb
  1. 测试
cp -r /usr/src/cudnn_samples_v8/ ~
cd ~/cudnn_samples_v8/mnistCUDNN/
make clean && make
./mnistCUDNN

make clean && make的结果
在这里插入图片描述
./mnistCUDNN指令结果如果是**Test passed!**则cudnn也安装成功!

# 查看cuda是否安装了正确版本
dpkg -l |grep cuda

在这里插入图片描述

如果安装错了删除cudnn

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*
sudo rm -rf /usr/include/cudnn.h
sudo rm -rf /usr/lib/x86_64-linux-gnu/libcudnn*

安装anaconda

参考教程

  1. 安装anaocnda
    release github
    国内下载,然后运行bash
bash Anaconda3-2020.07-Linux-x86_64.sh

vim ~/.bashrc
export PATH=$PATH:【你的安装目录】
source ~/.bashrc

conda --version
  1. 换国内的源
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/pytorch/ 
conda config --set show_channel_urls yes
  1. 创建一个单独的环境
conda create -n pytorch_gpu python=3.6
# 并且激活
source activate pytorch_gpu

conda install scikit-image scipy numba pillow matplotlib numpy
pip install fire tensorboardX protobuf opencv-python
  1. 安装pytorch
    查看版本
pip3 install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html

在线安装会很慢,可能会安装失败,如果失败就离线安装,下载

pip install torch-1.9.0+cu111-cp38-cp38-linux_x86_64.whl
pip install torchvision-0.10.0+cu111-cp38-cp38-linux_x86_64.whl
  1. 检查pytorch是否可用
    运行python3
>>> import torch
>>> print(torch.__version__)
1.9.1
>>> print(torch.cuda.is_available())
True
# 说明pytorch安装完成

conda 指令

conda 指令参考

conda create -n myenv python=3 # 创建一个名为myenv的环境并指定python版本为3
conda remove -n myenv --all  # 删除环境及下属所有包

activate myenv # 切换到myenv环境
deactivate myenv # 推出环境myenv
activate # 进入默认的base环境

conda env list # 列出所有环境
conda list # 列出当前环境的所有包

# 以pandas为例,安装,卸载,更新
conda search pandas # 查找可安装的包
conda install pandas # 安装包
conda remove pandas # 卸载包
conda update pandas # 更新包

#cuda,cudnn 在conda环境里
conda install cudatoolkit=11.1 
conda install cudnn=8.1.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge

# 删除环境
conda remove -n your_env_name --all

Cmake安装

  1. 下载 xx.tar.gz压缩包
    CMake
  2. 安装
$tar -zxvf xx.tar.gz
$./bootstrap
$make
$make install

安装中间出现的错误及解决

在这里插入图片描述

sudo apt-get install libssl-dev

在这里插入图片描述

sudo make install

安装spconv

spconv github

git clone https://github.com/traveller59/spconv.git --recursive
sudo apt-get install libboost-all-dev
cd spconv
python setup.py bdist_wheel
cd ./dist
pip install spconv-1.2.1-cp38-cp38-linux_x86_64.whl

ERROR

在这里插入图片描述
新版本cudnn的版本信息包含在cudnn_version.h中,而不是cudnn.h中
ERROR问题是识别不到版本,说明没有把版本头文件加入到本地的cuda/include中

sudo cp cuda/include/cudnn_version.h /usr/local/cuda/include

测试spconv

python
>>import spconv

安装OpenPCDet

git clone https://github.com/open-mmlab/OpenPCDet.git
cd OpenPCDet
pip install -r requirements.txt 
python setup.py develop

ERROR:GPU算力与pytorch不匹配

前CUDA算力高于当前PyTorch版本支持的CUDA算力的问题
nvcc fatal : Unsupported gpu architecture ‘compute_86
但我用网上的方法(export TORCH_CUDA_ARCH_LIST=“8.0”),不能解决,export这行代码对我的机器没有任何影响
给出我的方法成功了

sudo TORCH_CUDA_ARCH_LIST="8.0" python setup.py develop

安装mayavi 4.7.3

conda install vtk=8.2.0
pip install mayavi

跑DEMO

去gihub上下在一个.pth与训练文件,然后按下面步骤跑一下demo

cd OpenPCDet/tools
python demo.py --cfg_file cfgs/kitti_models/pv_rcnn.yaml     --ckpt pv_rcnn_8369.pth     --data_path ${
    POINT_CLOUD_DATA}

ERROR

在这里插入图片描述
github 官方解释: This results from an incompatibility between the recently released Traits 6.3.0 and the most recent release of Mayavi. Either upgrade to Traits 6.3.1, or use Traits 6.2.0 for now.
改成6.2.0有效

pip uninstall traits
pip install traits=6.2.0

成功显示

在这里插入图片描述

  • 2
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值