failed call to cuInit: CUDA_ERROR_NO_DEVICE

[wangames]

when I try run python train.py , it get an error :failed call to cuInit: CUDA_ERROR_NO_DEVICE

[YurongChen1998]

Hi, I got the same problem, do you fix it?

[wangames]

@DanielChen98 I have not solved it, but I find a repository that are more helpful and robust, https://github.com/you359/Keras-FasterRCNN, it works well on my machine。

[YurongChen1998]

@DanielChen98 I have not solved it, but I find a repository that are more helpful and robust, https://github.com/you359/Keras-FasterRCNN, it works well on my machine。

I have solved it.
change ./tools/train.py
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
to
os.environ["CUDA_VISIBLE_DEVICES"] = "0"