Skip to content

GCP Quickstart

Glenn Jocher edited this page Nov 30, 2019 · 24 revisions

1. Create VM

Select a PyTorch >=1.2 image from the GCP marketplace, check 'install GPU driver automatically', and select a 100 GB SSD for sufficient I/O speed. All dependencies are included in the preinstalled Anaconda distribution. Install Nvidia drivers if prompted to do so when first starting.

Screen Shot 2019-11-23 at 11 16 01 AM

2. Setup VM

Clone repo:

git clone https://github.com/ultralytics/yolov3

Download COCO dataset:

bash yolov3/data/get_coco_dataset_gdrive.sh

3. Start

cd yolov3 and:

  • Train: python3 train.py
  • Test: python3 test.py
  • Detect: python3 detect.py
Clone this wiki locally