Date: May 2, 2021

Instance type used: p3.8xlarge.

AMI used: AWS Deep Learning AMI (Ubuntu 18.04) (version when writing this note: 42.1).

Commands run on the machine to install & configure Detectron2 with CUDA:

conda activate pytorch_latest_p37

# There is a bug in the latest pytorch that affects D2's training, so downgrade it first
# [<https://github.com/facebookresearch/detectron2/issues/2837>](<https://github.com/facebookresearch/detectron2/issues/2837>)
pip install torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2

# Install for logging visualizations with Wandb
pip install wandb

# Clone
git clone <https://github.com/facebookresearch/detectron2.git>
# Remove black requirement, causes problems during build
sed -i '/black==/d' detectron2/setup.py
python -m pip install -e detectron2