keras-tcn does not seem to use GPU !!

philipperemy / keras-tcn

Keras Temporal Convolutional Network.

MIT License

1.87k stars 454 forks source link

keras-tcn does not seem to use GPU !! #167

Closed scstu closed 3 years ago

scstu commented 3 years ago

I'm reporting this Issue while using Colab. After installing Keras-tcn (using !pip install keras-tcn), everything is just working fine. however, every other layer is fast except when integrating tcn into the network, the code went very slow. btw, I'm using the pro subscription of google colab!.

philipperemy commented 3 years ago

@scstu not sure here because it works on GPU when not using collab (at least). So maybe check that you have installed tensorflow-gpu! It works with 2.3.0 (at least).