code does not see gpu

[skywo1f]

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]

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