I am trying to make a prediction using Tensorflow Object Detection API on Google COLAB. Already I successfully completed the train
I could fix this issue by using below python code in Google Colab note
import os os.environ["CUDA_VISIBLE_DEVICES"] = "1"