-
-
Notifications
You must be signed in to change notification settings - Fork 3.4k
GCP Quickstart
To get started using this repo quickly using a Google Cloud Platform (GCP) Deep Learning Virtual Machine (VM) follow the instructions below. New GCP users are eligible for a $300 free credit offer. Other quickstart options for this repo include our Jupyter Notebook and our Docker image at https://hub.docker.com/r/ultralytics/yolov3 .
Select a Deep Learning VM from the GCP marketplace, select an n1-standard-8 instance (with 8 vCPUs and 30 GB memory), add a GPU of your choice, check 'Install NVIDIA GPU driver automatically on first startup?', and select a 300 GB SSD Persistent Disk for sufficient I/O speed, then click 'Deploy'. All dependencies are included in the preinstalled Anaconda Python environment.
Clone this repo and install requirements.txt dependencies, including Python>=3.8 and PyTorch>=1.7.
$ git clone https://github.com/ultralytics/yolov3 # clone repo
$ cd yolov3
$ pip install -r requirements.txt # install dependencies
-
Train:
$ python train.py
-
Test:
$ python test.py
-
Detect:
$ python detect.py
Add 64GB of swap memory (to --cache
large datasets).
sudo fallocate -l 64G /swapfile
sudo chmod 600 /swapfile
sudo mkswap /swapfile
sudo swapon /swapfile
free -h # check memory
Mount local SSD
lsblk
sudo mkfs.ext4 -F /dev/nvme0n1
sudo mkdir -p /mnt/disks/nvme0n1
sudo mount /dev/nvme0n1 /mnt/disks/nvme0n1
sudo chmod a+w /mnt/disks/nvme0n1
cp -r coco /mnt/disks/nvme0n1
© 2024 Ultralytics Inc. All rights reserved.
https://ultralytics.com