tapnet icon indicating copy to clipboard operation
tapnet copied to clipboard

Published 20 hours ago • verified publisher icon google-deepmind

Is a GPU version of TF implied in order to run `live_demo.py` on a GPU?

Open weirenorweiren opened this issue 5 months ago • 4 comments

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?

weirenorweiren avatar Sep 08 '24 06:09 weirenorweiren