запуск train на gpu

ria-com / nomeroff-net

Nomeroff Net. Automatic numberplate recognition system.

GNU General Public License v3.0

459 stars 159 forks source link

Closed prozaklob closed 4 years ago

prozaklob commented 4 years ago

доброго времени суток! установлен tensorflow-gpu

как реализовать mrcnn_train на gpu?

запускается и насилует все равно проц.

gtx 1080 cuda 10.1 python 3.7

prozaklob commented 4 years ago

tensorflow==1.15.2 tensorflow-gpu==1.15.2 cudNN 10.0 (можно в папку с 10.1 закинуть библиотеки от 10) Все заработало