ubuntu下的openpcdet环境配置

nvidia 510 + cuda10.2 + cudnn + pytorch1.10

1 ubuntu换国内源(pip)

Ubuntu20.04下深度学习环境配置(持续维护)_伊文111的博客-CSDN博客_ubuntu深度学习环境搭建

2 安装gpu驱动

用ubuntu自带的驱动安装,“软件与更新”->“附近驱动”->选择推荐版本->“应用更改

”->完成后重启系统,调用nvidia-smi命令,会显示版本等信息,安装gpu驱动完成

2 安装anaconda(如博客1)

3 conda虚拟环境下配置OpenPcdet环境(初次使用conda环境并进行配置)

ubuntu 18.04 从零开始复现 OpenPCDet 训练kitti数据集并评估 可视化 损失函数_想Rui的猪的博客-CSDN博客

(1)安装cuda10.2(博客1)

进入官网,找对应版本不用太高,下载并run镜像文件(安装好后可删除run文件)

ubuntu下的openpcdet环境配置_第1张图片

 ubuntu下的openpcdet环境配置_第2张图片

 获取run镜像文件到本地 ,创建conda虚拟环境,执行run文件进行安装

wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run

conda create -n open python=3.7

sudo sh cuda_10.2.89_440.33.01_linux.run

run出现问题,cuda10.2不支持gcc-9,解决办法如下博客

Ubuntu20.04安装cuda10.2和cudnn7.6.5_小小小白~~的博客-CSDN博客

ubuntu下的openpcdet环境配置_第3张图片

continue->accept.然后将Driver前面的选项按回车,去掉;其他的保持不动,选择“Install”,稍等几分钟就装好了,会出现以下的界面。

ubuntu下的openpcdet环境配置_第4张图片

 cuda默认安装在/usr/local目录下 

配置环境

$sudo vim ~/.bashrc \\进入vim界面。输入字母i,进入编辑模式
\\在bashrc文件中输入以下命令,注意修改你的cuda版本
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.2/lib64
export PATH=$PATH:/usr/local/cuda-10.2/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.2
\\输入完成后,点击esc键并输入:wq!,再按esc键退出vim。

\\这时候返回终端了
$source ~/.bashrc \\运行.bashrc文件

nvcc --version

ubuntu下的openpcdet环境配置_第5张图片

(2)cudnn

https://developer.nvidia.com/rdp/cudnn-download

下载tar文件,解压到别的路径

 tar -xvf cudnn-linux-x86_64-8.4.0.27_cuda10.2-archive.tar.xz  -C /home/root123/software/

ubuntu下的openpcdet环境配置_第6张图片

 在conda环境中执行如下命令

 conda activate open

cp include/cudnn*.h /usr/local/cuda/include/
cp lib/libcudnn* /usr/local/cuda/lib64/
chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

ubuntu下的openpcdet环境配置_第7张图片

查看是否成功

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

ubuntu下的openpcdet环境配置_第8张图片

(3)安装pytorch-gpu版本1.10(用conda安装,添加配置conda国内源)

在./condarc文件加过一次就行(多个环境共用)

conda安装pytorch出现:CondaHTTPError: HTTP 000 CONNECTION FAILED for url……_han_hhh的博客-CSDN博客

Previous PyTorch Versions | PyTorch

conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 

 ubuntu下的openpcdet环境配置_第9张图片

 验证cuda cudnn

ubuntu下的openpcdet环境配置_第10张图片

 下面配置openpcdet

(4)安装spconv

pip install spconv-cu102 -i https://pypi.douban.com/simple

ubuntu下的openpcdet环境配置_第11张图片

ubuntu下的openpcdet环境配置_第12张图片

验证spconv是否安装成功

(4)下载openpcdet

(5)安装依赖项

pip install -r requirements.txt -i https://pipy.douban.com/simple

ubuntu下的openpcdet环境配置_第13张图片

(6)编译pcdet库

python setup.py develop

ubuntu下的openpcdet环境配置_第14张图片

此处报错:error: [Errno 2] No such file or directory: ':/usr/local/cuda-11.1:/usr/local/cuda-10.2/bin/nvcc': ':/usr/local/cuda-11.1:/usr/local/cuda-10.2/bin/nvcc',解决办法如下

sudo gedit ~/.bashrc

将export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.2

改为export CUDA_HOME=/usr/local/cuda-10.2

source ~/.bashrc

配置数据集(先下载kitti数据集到data),再用如下命令生成pkl文件

python -m pcdet.datasets.kitti.kitti_dataset create_kitti_infos tools/cfgs/dataset_configs/kitti_dataset.yaml

ubuntu下的openpcdet环境配置_第15张图片

ubuntu下的openpcdet环境配置_第16张图片 ubuntu下的openpcdet环境配置_第17张图片

 配置参数,训练模型

你可能感兴趣的:(深度学习,人工智能)