OmniDrive(2):环境搭建
文章目录
-
- 1. 环境安装
-
- 1.1 cuda 和cudnn安装
- 1.2 依赖包安装
- 2. 准备数据集及权重
-
- 2.1 下载nuScenes数据
- 2.2 下载预训练权重
- 3. 模型训练及推理
-
- 3.1 训练
- 3.2 推理
根据OmniDrive 官方的环境配置 Environment Setup
python 3.9
ubantu 22.04
cuda 11.7
torch1.13.1
1. 环境安装
1.1 cuda 和cudnn安装
访问 NVIDIA CUDA Toolkit 下载页面：CUDA Toolkit 11.7 Download。
选择以下选项：
-
Operating System: Linux
-
Ar