1. 环境准备:从零搭建Python与CUDA基础
搞3D目标检测最头疼的就是环境配置,特别是CUDA、cuDNN和PyTorch的版本匹配问题。我去年在部署CenterFusion时,光是环境配置就折腾了三天。下面分享经过实战验证的配置方案,帮你避开90%的坑。
1.1 Conda环境搭建技巧
首先用Miniconda创建独立环境,这是避免依赖冲突的关键。实测Python 3.7与PyTorch 1.7的兼容性最好:
bash复制conda create -n centerfusion python=3.7 -y
conda activate centerfusion
注意:如果系统默认Python版本不是3.7,需要先通过
conda install python=3.7降级。遇到过conda自动升级Python导致环境崩溃的情况,建议锁定版本:
bash复制conda config --set auto_update_conda false
1.2 CUDA与cuDNN精准匹配
CUDA 11.0 + cuDNN 8.0.4是经过验证的黄金组合。这里有个隐藏坑点:conda默认安装的cuDNN可能不带补丁版本号,导致后续DCNv2编译失败。推荐用nvidia官方channel安装:
bash复制conda install -c nvidia cudatoolkit=11.0 cudnn=8.0.4
安装后验证环境变量是否生效:
bash复制nvcc --version # 应显示CUDA 11.0
cat $(dirname $(which python))/../etc/conda/activate.d/cudnn_activate.sh # 检查cuDNN路径
1.3 PyTorch定制安装
官方pip源可能下载超时,推荐使用清华镜像源。特别注意要带+cu110后缀:
bash复制pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -i https://pypi.tuna.tsinghua.edu.cn/simple
验证安装时别
