Windows 10 RTX 2070 с использованием Keras GPU с Anaconda оптимизировать? - PullRequest
0 голосов
/ 19 мая 2019

Недавно я получил новый компьютер с Windows 10 и RTX 2070. Я установил anaconda, чтобы использовать python и фреймворки глубокого обучения, доступные как keras.Я устанавливаю с помощью навигатора anaconda пакет keras-gpu.Похоже, что установка этого пакета приведет к установке пакетов «cuda-toolkit 10» и «cudnn» на anaconda.

Мне было интересно, будет ли мой GPU оптимизирован использоваться во время тренировки на керасах.Фактически, в прошлом, когда я устанавливал keras gpu, мне приходилось самостоятельно устанавливать microsoft community 2015 и cuda toolkit 9.0 / Cudnn, чтобы заставить работать keras gpu.Так что, немного странно, что у меня не было ошибки.

Спасибо за помощь!

1 Ответ

1 голос
/ 19 мая 2019

Это зависит от того, какие бэкэнды использует ваша кераса. например Если вы используете tenorflow, следующее утверждение даст вам ответ.

print(tf.test.is_gpu_available())
...