google-deepmind / google-deepmind/tapnet
Is a GPU version of TF implied in order to run `live_demo.py` on a GPU?
Open
- Dominant language
- Jupyter Notebook
- Stars
- 2k
- Forks
- 192
- PR merge metrics
- No merged PRs in 30d
Description
I want to run `live_demo.py` on a GPU in Windows. The provided requirement files involves `jax` and `tensorflow` but `live_demo.py` only explicitly imports `jax`. Is a GPU version of `tensorflow` implied in order to run `live_demo.py` or `jax` on a GPU? Actually I am wondering why `tensorflow` is required in the requirements?
Contributor guide
Assessment
This issue has not been assessed yet.