Skip to content

Latest commit

 

History

History
35 lines (26 loc) · 1008 Bytes

README.md

File metadata and controls

35 lines (26 loc) · 1008 Bytes
Raw

Instructions

Requirements

  • NVIDIA GPU with the latest driver installed
  • docker / nvidia-docker

This build has been tested with Nvidia Drivers 460.91.03 and CUDA 11.2 on a Ubutun 20.04 machine. Please update the base image if you plan on using older versions of CUDA.

Build

Build the image with:

docker build -t project-dev -f Dockerfile .

Create a container with:

docker run --gpus all -v <PATH TO LOCAL PROJECT FOLDER>:/app/project/ --network=host -ti project-dev bash

and any other flag you find useful to your system (eg, --shm-size).

Set up

Once in container, you will need to auth using:

gcloud auth login

Debug

  • Follow this tutorial if you run into any issue with the installation of the tf object detection api

Updating the instructions

Feel free to submit PRs or issues should you see a scope for improvement.