Mac M1 conda env

[edwhu]

Hello,

I wanted to get this running on my M1 laptop for local development. The installation process deviated quite a bit from the README.

In particular, I had to setup tensorflow and jax differently than the README workflow. It went something like this:

conda create -n dreamerv3 python=3.8
conda activate dreamerv3

conda install -c apple tensorflow-deps
conda install jax -c conda-forge
pip install tensorflow-macos tensorflow_probability

pip install -r requirements.txt # comment out tensorflow-cpu in requirements.txt first.

I verified this environment by running example.py and by repeating the whole process again (conda installation and running example.py).

To make things easier for future users, I exported the conda env file. Then they don't need to manually install the conda dependencies and edit the requirements.txt.

Perhaps this will be useful for others?