Skip to content

GCP Quickstart

Glenn Jocher edited this page Nov 23, 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. Screenshot 2019-03-25 at 16 57 32

2. Setup VM

Clone repo:

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

Download COCO dataset:

bash yolov3/data/get_coco_dataset_gdrive.sh

Install pycocotools (optional):

git clone https://github.com/cocodataset/cocoapi && cd cocoapi/PythonAPI && make && cd ../.. && cp -r cocoapi/PythonAPI/pycocotools yolov3

3. Start

cd yolov3 and:

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