no CUDA-capable device is detected

DetectionTeamUCAS / FPN_Tensorflow

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

MIT License

348 stars 132 forks source link

Closed BigWilky closed 5 years ago

BigWilky commented 5 years ago

tf.version=1.13.1 CUDA Version 10.0.130 cudnn=7.5 GPU=RTX2080 我在训练的时候发现模型使用的是cpu进行的训练,并没有用到gpu，我再配置里需要进行怎么样的配置才可以转换为GPU模式？

yangxue0827 commented 5 years ago

cfgs.py中的GPU GROUP参数 @BigWilky

BigWilky commented 5 years ago

成了！！非常感谢！！