search

Walleclipse / Deep_Speaker-speaker_recognition_system

Keras implementation of ‘’Deep Speaker: an End-to-End Neural Speaker Embedding System‘’ (speaker recognition)
245 stars 81 forks source link

GPU版本 #32

Closed WenhaoSong closed 5 years ago

WenhaoSong commented 5 years ago

你好，请问这套代码有GPU版本的吗，现在在CPU上跑（渣处理器）出结果很慢~

Walleclipse commented 5 years ago

你好，这个代码就是gpu版的。你需要安装tensorflow-gpu和keras。

WenhaoSong commented 5 years ago

额 都装了的 但是并没有在GPU上跑~
我是采用的CUDA8.0可对应版本的Cudnn,TF... 请问你的CUDA,TF,Keras版本是多少啊，

Walleclipse commented 5 years ago

CUDA 8.0 应该可以的，你验证一下你是否是tensorflow-gpu版本。如果有tensorflow需要uninstall，重新安装tensorflow-gpu