MagicDrive环境配置（亲测跑通推理和训练）

起酥~

已于 2024-05-08 09:49:35 修改

阅读量820

点赞数 4

分类专栏：自动驾驶世界模型文章标签：自动驾驶

于 2024-05-07 17:34:59 首次发布

本文链接：https://blog.csdn.net/qq_42390752/article/details/138539067

版权

基础环境安装（Ampere GPU）

阅读github文件：因为本人在Ampere架构的GPU上安装环境，所以直接阅读doc/FAQ.md，根据Q2安装。

结束后安装xformer：

cd /root/miniconda3/envs/magicdrive/lib/python3.8/subprocess.py
#修改：line425, change check=True to check=False
pip install -e .

至此，可运行demo/run.py和demo/interactive_gui.py。

终端运行demo/interactive_gui.py时，可能出现浏览器无法访问本地路径的问题，用vscode连接后运行解决。（原因未知，可能我使用docker导致）

训练模型之前需要修改以下代码

cd magicdrive/dataset/nuscenes_dataset.py
#line110附近
#将info = self.data_infos[index]改为：
 try:
    info = self.data_infos[index]
 except:
    return None

cd /root/miniconda3/envs/magicdrive/lib/python3.8/site-packages/packaging/version.py
#line 197附近，修改match = self._regex.search(version)为：
if version=='0.0.19+.