search

Rayhane-mamah / Tacotron-2

DeepMind's Tacotron-2 Tensorflow implementation
MIT License
2.27k stars 905 forks source link

wavenet not running on GPU #464

Open shashidhar2609 opened 4 years ago

shashidhar2609 commented 4 years ago

i am trying to run wavenet with the following config but i am unable to run. please suggest how can i make use of gpu,do i need to set any variable ?

System info Cuda configuration:Cuda8 and cuDNN 7.1.4

os:Ubuntu

conda environment info: Tensorflow gpu version: 1.10.0, Cuda tookkit: Cuda8 and cudnn 7.1.3

Gpu info product: GV100GL [Tesla V100 PCIe 32GB] vendor: NVIDIA Corporation physical id: 1