Installation

Requirement

Our experiments are conducted with PyTorch 2.0.1, CUDA 11.7, Ubuntu 22.04, and NVIDIA Tesla A100 (80 GB). For other requirements, please check TRAINING.md and SAMPLING.md.
Preparation

Clone the repository to your local directory.
```
git clone https://github.com/OpenDriveLab/Vista.git
```
We provide an example on nuScenes dataset for training and sampling. Before you start, make sure you have:
- Downloaded the translated action annotations from here and put the JSON files into annos.
- Downloaded all splits of Trainval in Full dataset (v1.0) to your device following official instructions. After downloading, it should contain:
```
$<your-nusc-data-root>
├── samples
├── sweeps
├── ...
└── v1.0-trainval
```

We use conda to manage the environment.

conda create -n vista python=3.9 -y
conda activate vista

Install dependencies.

conda install -y pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia
pip3 install -r requirements.txt

=> Next: [Training]

Provide feedback