BEVFormer环境配置,无法安装detectron2

本文记录了作者在安装BEVFormer过程中遇到的CUDA版本不匹配问题,指出mmdet3d和detectron2对CUDA版本的要求不同,并提供了如何在不卸载旧版本CUDA情况下切换版本及避免硬件驱动冲突的解决方案。
AudioSeal 音频水印系统

**AudioSeal** 是 Meta 开源的语音水印系统,用于 AI 生成音频的检测和溯源。

前言

这个系列是记录我个人在学习过程中,遇到的一些问题,并记录我成功的方法,真实有效。

BEVFormer本身的安装文档非常不利好于新人,所以在此记录了我遇到的坑与解决办法,帮助和我一样初学入门BEV的朋友们。

使用平台

i7-12700k

4070ti

ubuntu20.04

conda虚拟环境中,python=3.8

遇到的问题

使用官方文档,甚至从 github 上完全下载了源代码之后直接 install,都始终报错 cuda 与 torch版本不对应。

以下是解决办法

这里的问题在于我们初学者最先接触的是 mmdet3d 的安装文档,出现这个问题的朋友们大多是先自己安装过一遍最新的 mmdet3d 后,再来学习 BEVFormer。

mmdet3d 中允许 cu <= cuda,也就是说在当时,我使用了 cu118 配合我下载的 cuda12.0 也是完全可以跑出 mmdet3d 给的 demo 的。

detectron2 中必须要求 cu == cuda, 这就说明了如果按照官方文档,我们必须使用 cuda 11.1。所以我们必须重新配置我们的 cuda 版本。

此说特别说明,可以不需要卸载之前的 cuda,只需要在 .bashrc 中对你所想要的 cuda 版本进行路径重新引导,然后 source ~/.bashrc 即可切换 cuda 版本

在安装新版本 cuda 时千万记得取消自动携带的显卡硬件驱动,否则重启后会黑屏。

推荐只用run的安装方式,(打脸以前的自己),这里在选择所需要的安装部分非常方便,可以自由勾选你所需要的部分,即可以很清晰的取消硬件驱动的安装。

即可成功解决

您可能感兴趣的与本文相关的镜像

AudioSeal 音频水印系统

AudioSeal 音频水印系统

语音合成
PyTorch
Cuda

**AudioSeal** 是 Meta 开源的语音水印系统,用于 AI 生成音频的检测和溯源。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值