How to change the code if I use CUDA 10？

DetectionTeamUCAS / FPN_Tensorflow

This is a tensorflow re-implementation of Feature Pyramid Networks for Object Detection.

https://github.com/DetectionTeamUCAS/FPN_Tensorflow

MIT License

347 stars 132 forks source link

How to change the code if I use CUDA 10？ #146

Closed HUI11126 closed 3 years ago

HUI11126 commented 3 years ago

2020-12-29 10:10:44.476602: E tensorflow/stream_executor/cuda/cuda_driver.cc:318] failed call to cuInit: CUDA_ERROR_NO_DEVICE: no CUDA-capable device is detected

HUI11126 commented 3 years ago

I solved the problem by changing _os.environ['CUDA_VISIBLEDEVICES'] = "0" in train.py