How to build RLDS on macOS

[jrabary]

Is there any documentation on how to install RLDS on macOS ?

[peterdavidfagan]

Hi @jrabary,

I was also interested in this but it doesn't look like there is a supported release on macOS (release file). The build files are in the source code but I guess it will take some work to parse through these and potentially make adjustments where needed; there might be fundamental challenges in supporting macOS.

[cemlyn007]

You can't install this because dm-reverb is only officially supported for Linux AMD64, but I do have a fork of dm-reverb where I did get it to compile to Mac ARM64. https://github.com/cemlyn007/reverb

[vyeevani]

The Bazel build has problems when using an unmodified version on a Mac. I ended up using a workaround where I just added this repo as a subtree to my work then pointed all my stuff at rlds.rlds allowing me to sidestep the Bazel install issues. Not the cleanest solution but the simplest :).

[peterdavidfagan]

Nice thanks @cemlyn007 I'll check out your fork next time I am looking to use rlds on MacOS.

@vyeevani do you have reference code for the above? Am I correct in my understanding, you are adding a submodule to your repo and then using relative imports that point towards the submodule files? Are there not issues related to the project dependencies (dm-reverb)?

[vyeevani]

I removed dm-reverb from the requirements. But your high level understanding is correct. If someone could update the bazel file to get that to work, it'd be greatly appreciated!

[vyeevani]

@peterdavidfagan I ended up needing the bazel to do a better compile. My recipe is super janky though.

1. build dm_reverb courtesy of @cemlyn007 's repo. I ended up having to do some other stuff on top of this to work with clang 15. You can find the exact changeset I used here: upgraded absl cemlyn007/reverb#1. The exact recipe that worked for me in terms of setting this up is:

conda create -n py39
conda activate py39
conda install python=3.9 -c conda-forge
pip install tensorflow-macos==2.16.1

2. apply this patch.txt

this worked for me but no guarantees that this will work for everyone

[peterdavidfagan]

Thanks for this @vyeevani