code does not see gpu - Githubissues

ManishSoni1908 / Mobilenet-ssd-keras

A Keras port of Single Shot MultiBox Detector as Mobilenet as a Backbone

62 stars 24 forks source link

code does not see gpu #10

Open skywo1f opened 4 years ago

skywo1f commented 4 years ago

If I run python and import tensorflow, it sees my gpu fine. But when I run training or inferencing with this code, it does not: 2020-01-19 11:16:48.866550: E tensorflow/stream_executor/cuda/cuda_driver.cc:318] failed call to cuInit: CUDA_ERROR_NO_DEVICE: no CUDA-capable device is detected

skywo1f commented 4 years ago

line ~22 of train does os.environ['CUDA_VISIBLE_DEVICES'] = "1" for some reason. Changing it to 0 fixes the problem