search

rlworkgroup / garage

A toolkit for reproducible reinforcement learning research.
MIT License
1.84k stars 309 forks source link

Garage without TF #2326

Closed aruhela closed 2 years ago

aruhela commented 2 years ago

Hi Garage Developers,

Does Garage work without TF? I would like to use just the Torch framework? Please let me know if and how it is possible?

More about the requirements In my miniconda env, TF takes 1.3 G space whereas Torch takes 1.4G space. Loading both of them for the very first time on HPC compute node from the Lustre filesystem takes 5minutes. Subsequent runs are extremely faster (4s due to the caching) but I would definitely like to optimize the first run as well.

Thanks, Amit

krzentner commented 2 years ago

Yes, Garage has supported being installed without tensorflow since PR #2304 . Note that this change is not yet present in any release, so you will need to install from source (e.g. pip install git+https://github.com/rlworkgroup/garage.git). The CI also does not test that this works, but I do intend to support it since I also dislike having to waste space on tensorflow. Do not do pip install garage[all] or pip install garage[tensorflow], since that will install tensorflow.

aruhela commented 2 years ago

Thanks KR, the above pip command worked very nicely.