Having Issues when I run train.py

EdjeElectronics / TensorFlow-Object-Detection-API-Tutorial-Train-Multiple-Objects-Windows-10

How to train a TensorFlow Object Detection Classifier for multiple object detection on Windows

Apache License 2.0

2.92k stars 1.3k forks source link

Having Issues when I run train.py #464

Open bhmin93 opened 4 years ago

bhmin93 commented 4 years ago

Hi, when I run: python3 model_main.py --logtostderr --train_dir=training/ --pipeline_config_path=training/ssdlite_mobilenet_v2_coco.config

I get this issue: self._session = tf_session.TF_NewSessionRef(self._graph._c_graph, opts) tensorflow.python.framework.errors_impl.InvalidArgumentError: device CUDA:0 not supported by XLA service while setting up XLA_GPU_JIT device number 0

Does anyone know how to solve this?

satyamedh commented 4 years ago

@bhmin93 , it is because ur GPU dosen't support Cuda, either u use cpu or buy a supported GPU