- NVIDIA GPU with the latest driver installed
- docker / nvidia-docker
Build the image with:
docker build -t project-dev -f Dockerfile.gpu .
Create a container with:
docker run -v <PATH TO LOCAL PROJECT FOLDER>:/app/project/ -ti project-dev bash
and any other flag you find useful to your system (eg, --shm-size).
Once in container, you will need to install gsutil, which you can easily do by running:
curl https://sdk.cloud.google.com | bash
Once gsutil is installed and added to your path, you can auth using:
gcloud auth login
- Follow this tutorial if you run into any issue with the installation of the tf object detection api